"""
'naive_bayes_demo_confusion_matrix.py'

Show the use of Naive Bayes Classifier
# Dr. Pan, Dept. of ECE, UAH

"""

import numpy as np
from sklearn.naive_bayes import GaussianNB
clf = GaussianNB()

from sklearn.datasets import load_iris
train_samples = load_iris()
X = train_samples.data
Y = train_samples.target

clf.fit(X, Y)

clf.score(X, Y)

# Verify the classification accuracy
Y_pred = clf.predict(X)

num_correct = np.sum(Y == Y_pred)
num_sample = np.size(Y_pred)
num_correct/num_sample

from sklearn.metrics import confusion_matrix,ConfusionMatrixDisplay
cm = confusion_matrix (Y,Y_pred)
disp = ConfusionMatrixDisplay(confusion_matrix = cm,display_labels = train_samples.target_names)
import matplotlib.pyplot as plt
disp.plot()
plt.show()

from sklearn.metrics import classification_report
print(classification_report (Y, Y_pred, target_names = train_samples.target_names))