bayes classifier on iris dataset

thebali-bayes-iris.py

from sklearn.naive_bayes import GaussianNB
import numpy as np
import pandas as pd
from csv import reader

filename = 'iris.data'
# url = "https://archive.ics.uci.edu/ml/machine-learning-databases/iris/iris.data"


# for row in reader(open(filename)):
# 	x = np.array(row)
# 	print(x)


irisdata = pd.read_csv(filename)
id1 = irisdata.iloc[:,0:4]
id2 = irisdata.iloc[:,[4]]

print(id1)
# print(id2)

bayesmodel = GaussianNB()
bayesmodel.fit(id1, id2.values.reshape(-1, 1))

prediction = bayesmodel.predict([2.0,6.0,3.3,4.0])

print(prediction)