aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--WMD_matching.py2
-rw-r--r--Wasserstein_Distance.py1
2 files changed, 1 insertions, 2 deletions
diff --git a/WMD_matching.py b/WMD_matching.py
index 8581ffe..2316a10 100644
--- a/WMD_matching.py
+++ b/WMD_matching.py
@@ -1,7 +1,5 @@
1import argparse 1import argparse
2import numpy as np 2import numpy as np
3from mosestokenizer import *
4import nltk
5import random 3import random
6from sklearn.feature_extraction.text import CountVectorizer, TfidfVectorizer 4from sklearn.feature_extraction.text import CountVectorizer, TfidfVectorizer
7from sklearn.preprocessing import normalize 5from sklearn.preprocessing import normalize
diff --git a/Wasserstein_Distance.py b/Wasserstein_Distance.py
index d8d08b8..925eca3 100644
--- a/Wasserstein_Distance.py
+++ b/Wasserstein_Distance.py
@@ -9,6 +9,7 @@ from sklearn.metrics.scorer import check_scoring
9from pathos.multiprocessing import ProcessingPool as Pool 9from pathos.multiprocessing import ProcessingPool as Pool
10from sklearn.metrics import euclidean_distances 10from sklearn.metrics import euclidean_distances
11import numpy as np 11import numpy as np
12from mosestokenizer import MosesTokenizer
12 13
13class Wasserstein_Matcher(KNeighborsClassifier): 14class Wasserstein_Matcher(KNeighborsClassifier):
14 """ 15 """