From b9c43ca1cc693326f583c301a3399cc28b9bb116 Mon Sep 17 00:00:00 2001 From: Adam Wojdyla Date: Mon, 30 May 2022 09:12:24 +0200 Subject: [PATCH] no changes --- run.py | 1 + 1 file changed, 1 insertion(+) 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)