Spaces:
Sleeping
Sleeping
Update model_methods.py
Browse files- model_methods.py +1 -1
model_methods.py
CHANGED
@@ -39,7 +39,7 @@ def predict(new_data):
|
|
39 |
cvec.fit(Z_train) # fit on lemmatized training data set
|
40 |
cvec.transform(Z_data) # transform new data
|
41 |
|
42 |
-
with open('
|
43 |
model = pickle.load(f)
|
44 |
pred = model.predict(Z_data)
|
45 |
return pred
|
|
|
39 |
cvec.fit(Z_train) # fit on lemmatized training data set
|
40 |
cvec.transform(Z_data) # transform new data
|
41 |
|
42 |
+
with open('final_model.sav','rb') as f:
|
43 |
model = pickle.load(f)
|
44 |
pred = model.predict(Z_data)
|
45 |
return pred
|