sklearn.mglearn 사용

import mglearn

 

model=KNeighborsClassifier(n_neighbors=1)
# n_neighbors=1
model.fit(X_train,y_train)

 

plt.figure(figsize=[10,8]) # 그래프 사이즈
mglearn.plots.plot_2d_classification(model,X_train,fill=True,eps=0.5,alpha=0.4) # 바탕색과 관련
mglearn.discrete_scatter(X_train[:,0],X_train[:,1],y_train) # train data에 대한 산점도
#mglearn.discrete_scatter(X_test[:,0],X_test[:,1],y_test) # test data에 대한 산점도

 

 

plt.figure(figsize=[10,8]) # 그래프 사이즈
mglearn.plots.plot_2d_classification(model,X_train,fill=True,eps=0.5,alpha=0.4) # 바탕색과 관련
#mglearn.discrete_scatter(X_train[:,0],X_train[:,1],y_train) # train data에 대한 산점도
mglearn.discrete_scatter(X_test[:,0],X_test[:,1],y_test) # test data에 대한 산점도