From 942d3eb835e56e870b2ad0c28014525c03747f71 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Niels=20M=C3=BCndler?= Date: Wed, 15 May 2019 00:00:15 +0200 Subject: [PATCH] Make use of externally defined joblib --- Pipfile | 1 + quantulum3/classifier.py | 2 +- requirements_classifier.txt | 1 + 3 files changed, 3 insertions(+), 1 deletion(-) 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