iris.py

#Import the Algorythm
from sklearn.neighbors import KNeighborsClassifier

# Instantiate the  Classifier
knn = KNeighborsClassifier()

# Train the Model
knn.fit(X,y)

# Create some test data , this corresponds with 0 species
test_data = [5.0 , 3.1,1.4,0.2]

# Predict using the trained Model
prediction = knn.predict([test_data])

print(prediction)