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 | """ |