[혼공머신] 2주차_knn 최적의 k값 찾기 시각화

본 게시글은 혼공학습단 11기로, 복습 차원에서 책 내용을 제 스타일대로 요약한 내용입니다.

#챕터3 #주니코니 #힘내랏

 

목차🐻

- 주차별 미션

- knn 최적의 k값 찾기_시각화

 

 

---

 

최적의_k_찾기__시각화_.ipynb

0.09MB

#최적의 k값 찾기(그래프로)
knr = KNeighborsRegressor()

x = np.arange(5,45).reshape(-1,1) #5~45까지 x 좌표 만들기 


for n in [1,5,10]: #k=1,5,10일때의 예측 그래프
  knr.n_neighbors = n #n_neighbors = 이웃의 개수 정하기  
  knr.fit(train_input, train_target)

  prediction = knr.predict(x)

  plt.scatter(train_input,train_target)
  plt.plot(x,prediction)
  plt.title('n_neighbors ={}'.format(n))
  plt.xlabel('length')
  plt.ylabel('weight')
  plt.show()