diff options
| -rw-r--r-- | Wasserstein_Distance.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/Wasserstein_Distance.py b/Wasserstein_Distance.py index 86204a1..925eca3 100644 --- a/Wasserstein_Distance.py +++ b/Wasserstein_Distance.py | |||
| @@ -9,7 +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 | from mosestokenizer import MosesTokenizer |
| 13 | 13 | ||
| 14 | class Wasserstein_Matcher(KNeighborsClassifier): | 14 | class Wasserstein_Matcher(KNeighborsClassifier): |
| 15 | """ | 15 | """ |
