knn做电影评分预测实现代码

以下是使用 Python 实现基于 KNN 算法的电影评分预测的示例代码：

import numpy as np
from sklearn.neighbors import NearestNeighbors

# 构造评分数据矩阵
# 行表示用户，列表示电影，每个元素表示评分（0表示未评分）
ratings = np.array([
    [5, 3, 0, 1, 4, 5],
    [4, 0, 4, 4, 3, 2],
    [0, 3, 3, 5, 4, 1],
    [5, 4, 3, 3, 0, 2],
    [3, 4, 5, 0, 5, 3],
    [2, 0, 1, 4, 0, 4]
])

# 要预测的用户和电影
target_user = 2
target_movie = 2

# 计算目标用户与其他用户的相似度
# 这里使用余弦相似度
def cosine_similarity(x, y):
    return np.dot(x, y) / (np.linalg.norm(x) * np.linalg.norm(y))

similarities = []
for i in range(ratings.shape[0]):
    if i == target_user:
        continue
    similarity = cosine_similarity(ratings[target_user], ratings[i])
    similarities.append((i, similarity))

# 按照相似度排序，并选择最相似的 K 个用户
K = 3
similarities.sort(key=lambda x: x[1], reverse=True)
neighbors = [sim[0] for sim in similarities[:K]]

# 计算 K 个用户对目标电影的评分的加权平均值
weights = [sim[1] for sim in similarities[:K]]
ratings_of_neighbors = ratings[neighbors, target_movie]
predicted_rating = np.dot(weights, ratings_of_neighbors) / np.sum(weights)

print("Predicted rating for user {} and movie {}: {:.2f}".format(target_user, target_movie, predicted_rating))

这个示例代码中使用了 `sklearn` 库中的 `NearestNeighbors` 类来计算 K 个最近邻居。如果你想自己实现 KNN 算法，可以使用 `numpy` 库来计算相似度和排序。