Model - Ipynb - Colaboratory

You might also like

Download as pdf or txt
Download as pdf or txt
You are on page 1of 3

Model.ipynb - Colaboratory https://colab.research.google.com/drive/1KxUpkQiYh...

import seaborn as sns
iris=sns.load_dataset('iris')

iris.info()

<class 'pandas.core.frame.DataFrame'>
RangeIndex: 150 entries, 0 to 149
Data columns (total 5 columns):
# Column Non-Null Count Dtype
--- ------ -------------- -----
0 sepal_length 150 non-null float64
1 sepal_width 150 non-null float64
2 petal_length 150 non-null float64
3 petal_width 150 non-null float64
4 species 150 non-null object
dtypes: float64(4), object(1)
memory usage: 6.0+ KB

iris.describe()

sepal_length sepal_width petal_length petal_width

count 150.000000 150.000000 150.000000 150.000000

mean 5.843333 3.057333 3.758000 1.199333

std 0.828066 0.435866 1.765298 0.762238

min 4.300000 2.000000 1.000000 0.100000

25% 5.100000 2.800000 1.600000 0.300000

50% 5.800000 3.000000 4.350000 1.300000

75% 6.400000 3.300000 5.100000 1.800000

max 7.900000 4.400000 6.900000 2.500000

iris['species'].unique()

array(['setosa', 'versicolor', 'virginica'], dtype=object)

from sklearn.model_selection import train_test_split
from sklearn.naive_bayes import GaussianNB

x =iris.drop(columns='species')
y=iris['species']

x_train,x_test,y_train,y_test=train_test_split(x,y,test_size=0.2)

model1=GaussianNB()

1 of 3 08/05/23, 10:03
Model.ipynb - Colaboratory https://colab.research.google.com/drive/1KxUpkQiYh...

 0s completed at 10:02 AM
NB_model =model1.fit(x_train,y_train)
y_pred=NB_model.predict(x_test)

y_pred

array(['setosa', 'versicolor', 'versicolor', 'virginica', 'setosa',


'virginica', 'virginica', 'setosa', 'virginica', 'virginica',
'versicolor', 'versicolor', 'setosa', 'virginica', 'setosa',
'setosa', 'setosa', 'setosa', 'versicolor', 'setosa', 'setosa',
'setosa', 'setosa', 'virginica', 'versicolor', 'versicolor',
'setosa', 'virginica', 'virginica', 'virginica'], dtype='<U10')

from sklearn.metrics import accuracy_score

print("Accuracy =",accuracy_score(y_test,y_pred))

Accuracy = 1.0

from sklearn.metrics import recall_score,precision_score

print("Precision =",precision_score(y_test,y_pred,average='micro'))

Precision = 1.0

from sklearn.metrics import classification_report

print(classification_report(y_test,y_pred))

precision recall f1-score support

setosa 1.00 1.00 1.00 13


versicolor 1.00 1.00 1.00 7
virginica 1.00 1.00 1.00 10

accuracy 1.00 30
macro avg 1.00 1.00 1.00 30
weighted avg 1.00 1.00 1.00 30

new_data =[[5.5,4.3,5.1,0.2],[4.5,2.3,2.1,0.2],[3.3,2.1,3.1,0.8]]

new_pred=NB_model.predict(new_data)

/usr/local/lib/python3.10/dist-packages/sklearn/base.py:439: UserWarning: X does n


warnings.warn(

new_pred

2 of 3 08/05/23, 10:03
Model.ipynb - Colaboratory https://colab.research.google.com/drive/1KxUpkQiYh...

array(['virginica', 'setosa', 'versicolor'], dtype='<U10')

new_data1 =[[5.5,4.3,5.1,0.2]]

new_pred1=NB_model.predict(new_data1)
new_pred1

/usr/local/lib/python3.10/dist-packages/sklearn/base.py:439: UserWarning: X does n


warnings.warn(
array(['virginica'], dtype='<U10')

new_pred1

array(['virginica'], dtype='<U10')

Colab paid products - Cancel contracts here

3 of 3 08/05/23, 10:03

You might also like