2주차 미션

# k-최근접 이웃 회귀 객체를 만든다
knr = KNeighborsRegressor()

# 5에서 45까지 x 좌표를 만듭니다
x = np.arange(5,45).reshape(-1, 1)

k = [1, 5, 10]

for n in k:
  #모델의 k 값을 변경합니다
  knr.n_neighbors = n
  # 모델 훈련
  knr.fit(train_input, train_target)

  # 지정한 범위 x에 대한 예측을 구합니다
  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()