Skip to content

Instantly share code, notes, and snippets.

@amankharwal

amankharwal/naive.py Secret

Created February 7, 2021 14:11
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save amankharwal/85ba27a5fcc6bd145ab8443b270d3e5b to your computer and use it in GitHub Desktop.
Save amankharwal/85ba27a5fcc6bd145ab8443b270d3e5b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
naive.py
from sklearn.naive_bayes import GaussianNB
from sklearn.naive_bayes import MultinomialNB
from sklearn import datasets
from sklearn.metrics import confusion_matrix
iris = datasets.load_iris()
gnb = GaussianNB()
mnb = MultinomialNB()
y_pred_gnb = gnb.fit(iris.data, iris.target).predict(iris.data)
cnf_matrix_gnb = confusion_matrix(iris.target, y_pred_gnb)
print(cnf_matrix_gnb)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment