Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -176,8 +176,8 @@ def main():
|
|
176 |
#loaded_vectorizer = dill.load(open('vectorizefile_classification.pickle', 'rb'))
|
177 |
#loaded_vectorizer = pickle.load(open('vectorizefile_classification.pickle', 'rb'))
|
178 |
|
179 |
-
|
180 |
-
|
181 |
pipeline_test_output = pipeline1.fit_transform(class_list)
|
182 |
#pipeline_test_output = loaded_vectorizer.transform(class_list)
|
183 |
predicted = loaded_model.predict(pipeline_test_output)
|
|
|
176 |
#loaded_vectorizer = dill.load(open('vectorizefile_classification.pickle', 'rb'))
|
177 |
#loaded_vectorizer = pickle.load(open('vectorizefile_classification.pickle', 'rb'))
|
178 |
|
179 |
+
from sklearn.pipeline import Pipeline
|
180 |
+
pipeline1 = Pipeline([('count_vect',count_vect),('tfidf_transformer',tfidf_transformer)])
|
181 |
pipeline_test_output = pipeline1.fit_transform(class_list)
|
182 |
#pipeline_test_output = loaded_vectorizer.transform(class_list)
|
183 |
predicted = loaded_model.predict(pipeline_test_output)
|