diff options
Diffstat (limited to 'Wasserstein_Distance.py')
| -rw-r--r-- | Wasserstein_Distance.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/Wasserstein_Distance.py b/Wasserstein_Distance.py index d8d08b8..86204a1 100644 --- a/Wasserstein_Distance.py +++ b/Wasserstein_Distance.py | |||
| @@ -9,6 +9,7 @@ from sklearn.metrics.scorer import check_scoring | |||
| 9 | from pathos.multiprocessing import ProcessingPool as Pool | 9 | from pathos.multiprocessing import ProcessingPool as Pool |
| 10 | from sklearn.metrics import euclidean_distances | 10 | from sklearn.metrics import euclidean_distances |
| 11 | import numpy as np | 11 | import numpy as np |
| 12 | from mosestokenizer import MosesDetokenizer | ||
| 12 | 13 | ||
| 13 | class Wasserstein_Matcher(KNeighborsClassifier): | 14 | class Wasserstein_Matcher(KNeighborsClassifier): |
| 14 | """ | 15 | """ |
