#machine-learning #pytorch #fast-ai
#машинное обучение #pytorch #быстрый ии
Вопрос:
У меня есть модель быстрой коллаборативной фильтрации искусственного интеллекта. Я хотел бы предсказать на основе этой модели новый кортеж. У меня возникли проблемы с функцией прогнозирования
Из их документации,
Signature: learn.predict(item, rm_type_tfms=None, with_input=False)
Docstring: Prediction on `item`, fully decoded, loss function decoded and probabilities
File: ~/playground/virtualenv/lib/python3.8/site-packages/fastai/learner.py
Type: method
Как мне определить элемент, который мне нужно передать. Допустим, для набора данных movielens, для пользователя, уже включенного в набор данных, мы хотели бы порекомендовать набор фильмов, как мы передаем идентификатор пользователя?
Я попытался дать здесь некоторый ответ — https://forums.fast.ai/t/making-predictions-with-collaborative-filtering/3900
learn.predict( [np.array([3])] )
Кажется, я получаю сообщение об ошибке: TypeError: list indices must be integers or slices, not list
Ответ №1:
Я думаю, это поможет:
https://medium.com/@igorirailean/a-recommender-system-using-fastai-in-google-colab-110d363d422f
Документация также содержит следующую информацию:
dl = learn.dls.test_dl(test_df)
learn.get_preds(dl=dl)
Это помогло мне.