"""
'naive_bayes_demo.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