NICT Club Blogs.

Naive Bayes Classifier

Sudeep Mishra PP
Sudeep Mishra

The Naive Bayes classifier is a probabilistic machine learning algorithm for classification. It calculates the probability of an instance belonging to a class based on observed features, assuming independence among features. Despite its simplicity, Naive Bayes is often effective and computationally efficient for various classification tasks.

Source code:

#Diabetes Prediction Using Naive Bayes Classifier

import pandas as pd

from sklearn import metrics

from sklearn.naive_bayes import GaussianNB

dataset = pd.read_csv('Diabetes.csv')

split = int(len(dataset)*0.7)

train, test = dataset[:split], dataset[split:]

p = train['Pragnency'].values

g = train['Glucose'].values

bp= train['Blod Pressure'].values

st= train['Skin Thikness'].values

ins= train['Insulin'].values

bmi= train['BMI'].values

dfp= train['DFP'].values

a= train['Age'].values

d= train['Diabetes'].values

trainfeatures=zip(p,g,bp,st,ins,bmi,dfp,a)

traininput=list(trainfeatures)

#print("traininput:")

#print(traininput)

model = GaussianNB()

model.fit(traininput,d)

p = test['Pragnency'].values

g = test['Glucose'].values

bp= test['Blod Pressure'].values

st= test['Skin Thikness'].values

ins= test['Insulin'].values

bmi= test['BMI'].values

dpf= test['DFP'].values

a= test['Age'].values

d= test['Diabetes'].values

testfeatures=zip(p,g,bp,st,ins,bmi,dpf,a)

testinput=list(testfeatures)

predicted= model.predict(testinput)

print("Actual Class: ", *d)

print("Predicted Class:", *predicted)

print("Confusion Matrix")

print(metrics.confusion_matrix(d, predicted))

print("********Classifiaction Measures*********")

print("Accuracy:",metrics.accuracy_score(d,predicted))

print("Recall:",metrics.recall_score(d,predicted))

print("Precision:",metrics.precision_score(d,predicted))

print("F1-Score:",metrics.f1_score(d,predicted))

Download CSV File


More Stories