From 4cb6986480def9b0c91fb46e276839c60f96aa49 Mon Sep 17 00:00:00 2001 From: Yigit Sever Date: Sat, 21 Sep 2019 15:06:32 +0300 Subject: Tidy up Wasserstein classes into one file --- WMD_matching.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'WMD_matching.py') 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 import random from sklearn.feature_extraction.text import CountVectorizer, TfidfVectorizer from sklearn.preprocessing import normalize -from Wass_Matcher import Wasserstein_Matcher +from Wasserstein_Distance import Wasserstein_Matcher def load_embeddings(path, dimension=300): """ -- cgit v1.2.3-70-g09d2