Fastai 预测协作学习模型

如何解决Fastai 预测协作学习模型

我有一个 Fast ai 协同过滤模型。我想在这个模型上预测一个新的元组。 我在使用预测功能时遇到问题

来自他们的文档，

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

如何定义需要传递的Item。假设对于一个movielens数据集，对于已经在数据集中的用户，我们想推荐一组电影，我们如何传递userID？

我试图在这里遵循一些答案 - 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

解决方法

我认为这会有所帮助：
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)

它帮助了我。