Replacing joblib library.

wisepythagoras · Jan 21, 2021 · 4cefa1f · 4cefa1f
1 parent 439ddf1
commit 4cefa1f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -33,7 +33,7 @@ torsocks curl -L http://httpbin.org/ip
 ## Required Python 3 Modules
 
 ``` bash
-pip install sklearn dpkt
+pip install sklearn dpkt joblib
 ```
 
 ## Data Collection

diff --git a/predict.py b/predict.py
@@ -4,7 +4,7 @@
 import sys
 import json
 import utils
-from sklearn.externals import joblib
+from joblib import load
 
 
 if len(sys.argv) == 1:
@@ -46,7 +46,7 @@
 print("Loading the classifier...")
 
 # Try to read the classifier.
-classifier = joblib.load("./classifier-nb.dmp")
+classifier = load("./classifier-nb.dmp")
 i = 0
 right = 0
 wrong = 0

diff --git a/utils.py b/utils.py
@@ -5,7 +5,7 @@
 from sklearn.naive_bayes import GaussianNB, BernoulliNB, MultinomialNB
 from sklearn.neural_network import MLPClassifier
 from sklearn.metrics import accuracy_score
-from sklearn.externals import joblib
+from joblib import dump
 
 
 def empty_csv():
@@ -177,7 +177,7 @@ def train(streams, labels):
     clf = clf.fit(training_x, training_y)
 
     # Save a snapshot of this classifier.
-    joblib.dump(clf, "./classifier-nb.dmp", compress=9)
+    dump(clf, "./classifier-nb.dmp", compress=9)
 
     # Get the prediction.
     predictions = clf.predict(testing_x)