Skip to content

Instantly share code, notes, and snippets.

@seanbenhur
Created May 15, 2021 12:12
Show Gist options
  • Save seanbenhur/60d19aa66392ce47ac6a20f3594aec4f to your computer and use it in GitHub Desktop.
Save seanbenhur/60d19aa66392ce47ac6a20f3594aec4f to your computer and use it in GitHub Desktop.
model = make_pipeline(TfidfVectorizer(), MultinomialNB())
model.fit(X_train, y_train)
y_pred = model.predict(X_test)
print(classification_report(y_test,y_pred))
#precision recall f1-score support
# 0 0.67 0.94 0.79 319
# 1 0.96 0.92 0.94 389
# 2 0.87 0.98 0.92 394
# 3 0.97 0.36 0.52 251
# accuracy 0.84 1353
# macro avg 0.87 0.80 0.79 1353
#weighted avg 0.87 0.84 0.82 1353
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment