naive bayes update

import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.naive_bayes import GaussianNB
from sklearn.metrics import accuracy_score,confusion_matrix
from sklearn.datasets import load_iris

iris=load_iris()
X=iris.data
y=iris.target

X_train,X_test,y_train,y_test=train_test_split(X,y,test_size=0.3)

model=GaussianNB()
model.fit(X_train,y_train)
y_pred=model.predict(X_test)

acc=accuracy_score(y_test,y_pred)
cm=confusion_matrix(y_test,y_pred)

print(acc)
print(cm)

Output 

Accuracy: 0.9555555555555556

Confusion Matrix:
[[17 0 0]
 [ 0 17 1]
 [ 0 1 9]]

Comments

Popular posts from this blog

Dijkstra's Algorithm Matrix

k means JDK java

Chi square test