KNN(K-Nearest Neighbors) [영화 평점 예측] (3)

KNN 회귀

 

KNN 회귀를 이용한 영화 평점 예측

평이 좋다” vs “평이 나쁘다” 레이블로 분류하는 게 아니라 실제 IMDb(인터넷 영화 데이터베이스) 등급(별점)을 예측

from sklearn.neighbors import KNeighborsRegressor

regressor = KNeighborsRegressor(n_neighbors=3, weights="distance")  # uniform: 단순평균(default) / distance: 가중평균

training_points = [
    [0.5, 0.2, 0.1],
    [0.9, 0.7, 0.3],
    [0.4, 0.5, 0.7]
]

training_labels = [5.0, 6.8, 9.0]
regressor.fit(training_points, training_labels)

unknown_points = [
    [0.2, 0.1, 0.7],
    [0.4, 0.7, 0.3],
    [0.4, 0.5, 0.7]
]

guesses = regressor.predict(unknown_points)
guesses		# array([7.28143288, 7.06939384, 9.        ])