6 Naive Bayesclassifn Algo
6 Naive Bayesclassifn Algo
[4]: data.head()
[5]: (150, 6)
[6] : data.head()
1
148 149 6.2 3.4 5.4 2.3
149 150 5.9 3.0 5.1 1.8
Species
145 Iris-virginica
146 Iris-virginica
147 Iris-virginica
148 Iris-virginica
[8] : 149 Iris-virginica
data.info()
<class
'pandas.core.frame.DataFrame'>
RangeIndex: 150 entries, 0 to 149
Data columns (total 6 columns):
# Column Non-Null Count Dtype
2
[11] : x = data.drop(['Species'], axis=1)
y = data.drop(['SepalLengthCm', 'SepalWidthCm',
'PetalLengthCm', ␣
𝗌'PetalWidthCm'], axis=1)
print(x)
print(y)
print(x.shape)
print(y.shape)
Id SepalLengthC SepalWidthC PetalLengthC PetalWidthC
m m m m
0 1 5.1 3.5 1.4 0.2
1 2 4.9 3.0 1.4 0.2
2 3 4.7 3.2 1.3 0.2
3 4 4.6 3.1 1.5 0.2
4 5 5.0 3.6 1.4 0.2
.. … … … … …
145 146 6.7 3.0 5.2 2.3
146 147 6.3 2.5 5.0 1.9
147 148 6.5 3.0 5.2 2.0
148 149 6.2 3.4 5.4 2.3
149 150 5.9 3.0 5.1 1.8
print(X_train.shape)
print(X_test.shape)
print(y_train.shape)
print(y_test.shape)
3
(120, 5)
(30, 5)
(120, 2)
(30, 2)
[14]: from sklearn.naive_bayes import GaussianNB
[15]: GaussianNB()
[15]: GaussianNB()