Как исправить функцию, в которой цикл for не повторяется. Система рекомендаций на Python ⇐ Python

[Anonymous]

Я работаю над системой рекомендаций с библиотекой Python Surprise.
Я пытаюсь создать функцию, которая перебирает полный список фильмов и прогнозирует их значение рейтинга по идентификатору пользователя, который определяется при вызове функции.

Это для Python3 с библиотекой сюрприз и алгоритмом SVD. Набор данных взят из MovieLens с открытым исходным кодом: ссылка на набор данных

Код: Выделить всё

from surprise import Dataset
from surprise import Reader
from surprise import SVD
import pandas as pd

reader = Reader(line_format='user item rating timestamp', sep='\t')
data = Dataset.load_from_file('./dataset/ml-100k/u.data', reader=reader)
movies = pd.read_csv('./dataset/ml-100k/u.item',sep='|', encoding='latin-1', usecols=[0,1], names=['iid', 'name'])
algorithm = SVD()
trainset = data.build_full_trainset()
algorithm.fit(trainset)

def prediction(uid):
ratings=[]
for iid in movies['iid']:
pred = algorithm.predict(uid, iid)
ratings.append(pred[3])
return ratings

prediction(321)

Я ожидаю, что функция построит список оценок (значение прогноза [3]) для каждого элемента и одного пользователя, указанного при вызове функции.

Код: Выделить всё

[3.52986,
3.69845,
2.98954,
3.00545,
3.84254,
...]

Но фактический результат представляет собой список длиной, равной длине списка фильмов (что нормально), но с одним и тем же рейтингом, повторяемым снова и снова.

Код: Выделить всё

 [3.52986,
3.52986,
3.52986,
3.52986,
3.52986,
3.52986,
3.52986,
3.52986,
...]

Поэтому, глядя на фактический результат, я думаю, что существует проблема с циклом for, который выполняет итерацию и добавляет результат, но использует один и тот же iid на каждой итерации. Я пытаюсь создать функцию, в которой uid остается фиксированным, но iid меняется для каждой итерации.

РЕДАКТИРОВАТЬ: как было предложено, я добавил print(iid) внутри цикла, и он выполняет итерацию правильно, печатая все iids. Но внутри строки pred = алгоритм.predict(uid, iid) это не работает

Подробнее здесь: https://stackoverflow.com/questions/578 ... -in-python