sklearn.externals 삭제

rickiepark · Jun 3, 2021 · f0f4471 · f0f4471
1 parent f6cf608
commit f0f4471
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/ml_editor/model_v2.py b/ml_editor/model_v2.py
@@ -2,7 +2,7 @@
 from pathlib import Path
 
 import spacy
-from sklearn.externals import joblib
+import joblib
 from tqdm import tqdm
 import pandas as pd
 import nltk
@@ -196,7 +196,7 @@ def get_pos_score_from_text(input_text):
     positive_proba = get_question_score_from_input(input_text)
     output_str = (
         """
-        Question score (0 is worst, 1 is best): 
+        Question score (0 is worst, 1 is best):
         <br/>
         %s
     """

diff --git a/ml_editor/model_v3.py b/ml_editor/model_v3.py
@@ -2,7 +2,7 @@
 from pathlib import Path
 
 import spacy
-from sklearn.externals import joblib
+import joblib
 from tqdm import tqdm
 import pandas as pd
 import nltk
@@ -94,7 +94,7 @@ def get_recommendation_and_prediction_from_text(input_text, num_feats=10):
      %s
     <br/>
     <br/>
-    
+
     Recommendations (ordered by importance):
     <br/>
     <br/>

diff --git a/tests/test_model.py b/tests/test_model.py
@@ -6,7 +6,7 @@
 import pytest
 
 # Needed for pytest to resolve imports properly
-from sklearn.externals import joblib
+import joblib
 
 from ml_editor.data_ingestion import parse_xml_to_csv
 from ml_editor.data_processing import (