KNN Classifier
KNN Classifier
irisData = load_iris()
print(irisData.DESCR)
data = pd.DataFrame(irisData.data)
data
data_2 = pd.DataFrame(irisData.target)
data_2
data_2.values
X = irisData.data
y = irisData.target
knn = KNeighborsClassifier(n_neighbors=5)
knn.fit(X_train, y_train)
pred= knn.predict(X_test)
error_rate = []
for i in range(1,40):
knn = KNeighborsClassifier(n_neighbors=i)
knn.fit(X_train,y_train)
pred_i = knn.predict(X_test)
error_rate.append(np.mean(pred_i != y_test))
plt.figure(figsize=(10,6))
plt.plot(range(1,40),error_rate,color='blue', linestyle='dashed', marker='o',
markerfacecolor='red', markersize=10)
plt.title('Error Rate vs. K Value')
plt.xlabel('K')
plt.ylabel('Error Rate')
knn = KNeighborsClassifier(n_neighbors=1)
knn.fit(X_train,y_train)
pred = knn.predict(X_test)
print('WITH K=1')
print('\n')
print(confusion_matrix(y_test,pred))
print('\n')
print(classification_report(y_test,pred))