嵌入NN fastai python

问题描述

https://github.com/TannerGilbert/Tutorials/blob/master/FastAI/Book%20Recommendation%20System.ipynb 我正在使用fastai，已经成功构建了模型，但是我不知道如何为用户设置个人推荐 使用EmbeddingNN模型

from pandas import DataFrame
import numpy as np

from fastai.collab import CollabDataBunch
from fastai.collab import collab_learner 
ratings=DataFrame(zip(a,b,c),columns =['partner_id','product_id','rating'])        
data = CollabDataBunch.from_df(ratings,seed=42,valid_pct=0.1,user_name='partner_id',item_name='product_id',rating_name='rating')
learn = collab_learner(data,use_nn=True,emb_szs={'partner_id': 40,'product_id':40},layers=[256,128],y_range=(1,5))
learn.load('C:\\test\\recomend-1')

尝试了以下选项： learn.predict() 和 learn.get_preds() 例如

test=learn.predict( ratings.partner_id["82"]) 导致： 在pandas._libs.index.IndexEngine.get_loc

中的文件“ pandas_libs \ index.pyx”，第135行

pandas._libs.index.Int64Engine._check_type中的文件“ pandas_libs \ index_class_helper.pxi”，第109行

KeyError：'82'

我是菜鸟，不知道如何使用预测。 帮帮我

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）