diff options
Diffstat (limited to 'WMD_matching.py')
-rw-r--r-- | WMD_matching.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/WMD_matching.py b/WMD_matching.py index 7fdf2f3..38dbff4 100644 --- a/WMD_matching.py +++ b/WMD_matching.py | |||
@@ -5,7 +5,7 @@ import nltk | |||
5 | import random | 5 | import random |
6 | from sklearn.feature_extraction.text import CountVectorizer, TfidfVectorizer | 6 | from sklearn.feature_extraction.text import CountVectorizer, TfidfVectorizer |
7 | from sklearn.preprocessing import normalize | 7 | from sklearn.preprocessing import normalize |
8 | from Wass_Matcher import Wasserstein_Matcher | 8 | from Wasserstein_Distance import Wasserstein_Matcher |
9 | 9 | ||
10 | def load_embeddings(path, dimension=300): | 10 | def load_embeddings(path, dimension=300): |
11 | """ | 11 | """ |