sklearn.knn.KNeiborsClassifier(K값 변화에 따른 score 관찰)

# 속성 컨트롤
col1=3
col2=1

X=iris.data[:,[col1,col2]]
y=iris.target

test_scores=[]
train_scores=[]
index=range(1,30)
X_train,X_test,y_train,y_test=train_test_split(X,y)

for k in index:
    model=KNeighborsClassifier(n_neighbors=k)
    model.fit(X_train,y_train)
    test_scores.append(model.score(X_test,y_test))
    train_scores.append(model.score(X_train,y_train))
#plt.title('K에 따른 score변화')    
plt.plot(index,test_scores,'ro:',label='test',alpha=0.5)
plt.plot(index,train_scores,'bo:',label='train',alpha=0.5)
plt.xlabel('k')
plt.ylabel('score',rotation=50)
plt.legend()