diff --git a/Pipfile b/Pipfile index 0dceef2..2b0e360 100644 --- a/Pipfile +++ b/Pipfile @@ -11,6 +11,7 @@ sklearn = "*" wikipedia = "*" stemming = "*" num2words = "*" +joblib = "*" [dev-packages] yapf = "*" diff --git a/quantulum3/classifier.py b/quantulum3/classifier.py index 9104073..de0adf6 100644 --- a/quantulum3/classifier.py +++ b/quantulum3/classifier.py @@ -14,7 +14,7 @@ try: from sklearn.linear_model import SGDClassifier from sklearn.feature_extraction.text import TfidfVectorizer - from sklearn.externals import joblib + import joblib USE_CLF = True except ImportError: SGDClassifier, TfidfVectorizer = None, None diff --git a/requirements_classifier.txt b/requirements_classifier.txt index c2da974..826dbc3 100644 --- a/requirements_classifier.txt +++ b/requirements_classifier.txt @@ -1,5 +1,6 @@ numpy scipy scikit-learn +joblib wikipedia stemming \ No newline at end of file