如何获得fastai v2文本列的预测分类

Question

我训练了一个模型来对文本进行分类。我想在生产中使用这种型号

为此，我已将其导出为pkl，然后加载load_learner：

learner = load_learner('/MyPath/MyNewModel.pkl', "MyNewModel.pkl")

然后，我已加载我的csv并提取包含要使用我的模型分类的文本的列：

df_loadNew = pd.read_csv('/My_data_to_tryout.csv')
df_loadNew=df_loadNew.iloc[:,:1]
df_loadNew.columns = ['text']

然后，我根据fastai v2文档创建了数据加载器：

real_dl=learner.dls.test_dl(df_loadNew['text'])

然后，我可以获得_preds：

pred_class,pred_idx =learner.get_preds(dl=real_dl)

我的问题是如何在与原始文本相同的数据帧中获得预测的分类