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, train..
2024. 3. 22.