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
Post a Comment