diff --git a/run.py b/run.py index 75de0c0..961fdac 100644 --- a/run.py +++ b/run.py @@ -15,6 +15,7 @@ x_train_vectorized = tfidf_vectorizer.fit_transform(x_train['text'].values) mnb_model = MultinomialNB().fit(x_train_vectorized, y_train.values.ravel()) + # Dev data x_dev_prepared = tfidf_vectorizer.transform(x_dev['text'].values) predictions = mnb_model.predict(x_dev_prepared)