s444507/paranormal-or-skeptic-ISI-public
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

no changes

Browse Source
This commit is contained in:
Adam Wojdyla 2022-05-30 09:12:24 +02:00
parent 2194c77c4b
commit b9c43ca1cc
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
run.py
View File

@ -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)