5/23/22, 8:08 PM Exp-8 - Jupyter Notebook

In [1]:

from sklearn.model_selection import train_test_split

from sklearn.naive_bayes import GaussianNB
from sklearn.datasets import load_iris

In [2]:

data = load_iris()

In [3]:

X, y = data['data'], data['target']

In [5]:

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

In [6]:

nb = GaussianNB()
nb.fit(X_train, y_train)

Out[6]:

GaussianNB()

In [7]:

nb.score(X_test, y_test)

Out[7]:

0.9473684210526315

localhost:8888/notebooks/Exp-8.ipynb 1/2
5/23/22, 8:08 PM Exp-8 - Jupyter Notebook

In [8]:

d = {k:data['target_names'][k] for k in range(3)}

for predicted, correct in zip(nb.predict(X_test), y_test):

print(f'Predicted {d[predicted]} for {d[correct]}')

Predicted virginica for virginica

Predicted virginica for virginica

Predicted versicolor for versicolor

Predicted versicolor for versicolor

Predicted setosa for setosa

Predicted virginica for virginica

Predicted setosa for setosa

Predicted versicolor for versicolor

Predicted setosa for setosa

Predicted setosa for setosa

Predicted setosa for setosa

Predicted virginica for virginica

Predicted virginica for virginica

Predicted setosa for setosa

Predicted setosa for setosa

Predicted virginica for virginica

Predicted virginica for virginica

Predicted setosa for setosa

Predicted virginica for virginica

Predicted versicolor for versicolor

Predicted virginica for virginica

Predicted setosa for setosa

Predicted virginica for virginica

Predicted virginica for versicolor

Predicted versicolor for versicolor

Predicted versicolor for virginica

Predicted setosa for setosa

Predicted versicolor for versicolor

Predicted setosa for setosa

Predicted versicolor for versicolor

Predicted versicolor for versicolor

Predicted virginica for virginica

Predicted virginica for virginica

Predicted virginica for virginica

Predicted setosa for setosa

Predicted setosa for setosa

Predicted versicolor for versicolor

Predicted versicolor for versicolor

In [ ]:

localhost:8888/notebooks/Exp-8.ipynb 2/2