Practical04.ipynb - Colab
Practical04.ipynb - Colab
ipynb - Colab
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.naive_bayes import GaussianNB
from sklearn.metrics import accuracy_score, classification_report
data = pd.read_csv('winequality_red.csv')
print(data.head())
alcohol quality
0 9.4 5
1 9.8 5
2 9.8 5
3 9.8 6
4 9.4 5
model = GaussianNB()
model.fit(X_train, y_train)
▾ GaussianNB
GaussianNB()
y_pred = model.predict(X_test)
Accuracy: 0.54