问题描述
我有一个 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)
它帮助了我。