[AI]/python.sklearn
sklearn.mglearn 사용
givemebro
2020. 4. 9. 11:26
반응형
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에 대한 산점도
반응형