Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from sklearn.model_selection import train_test_split
- from sklearn.neighbors import KNeighborsClassifier
- #from sklearn.naive_bayes import GaussianNB
- from sklearn import datasets
- iris=datasets.load_iris()
- print("Iris Data set loaded...")
- x_train, x_test, y_train, y_test = train_test_split(iris.data,iris.target,test_size=0.3)
- #random_state=0
- print("Dataset is split into training and testing samples...")
- print("Size of trainng data and its label",x_train.shape,y_train.shape)
- print("Size of trainng data and its label",x_test.shape, y_test.shape)
- for i in range(len(iris.target_names)):
- print("Label", i , "-",str(iris.target_names[i]))
- classifier = KNeighborsClassifier(n_neighbors=1)
- #gnb = GaussianNB()
- #y_pred = gnb.fit(x_train, y_train).predict(x_test)
- #print("Number of mislabeled points out of a total %d points : %d" %(x_test.shape[0], (y_test != y_pred).sum()))
- classifier.fit(x_train, y_train)
- y_pred=classifier.predict(x_test)
- print("Results of Classification using K-nn with K=1 ")
- for r in range(0,len(x_test)):
- print(" Sample:", str(x_test[r]), " Actual-label:", str(y_test[r]), " Predicted-label:", str(y_pred[r]))
- print("Classification Accuracy :" , classifier.score(x_test,y_test)*100);
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement