嵌入式快速python

我正在使用fastai，我已经成功构建了模型，但我不知道如何为用户设置个人推荐 利用嵌入神经网络模型

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\u preds（）

比如说

test=learn.predict（评级.合作伙伴id[“82”）

导致： 文件“pandas_libs\index.pyx”，第135行，在pandas._libs.index.IndexEngine.get_loc中

文件“pandas\libs\index\u class\u helper.pxi”，第109行，在pandas.\u libs.index.Int64Engine.\u check\u type中

KeyError:'82'

我是个笨蛋，不知道如何使用预测。 帮帮我