如何获得一列文本 fastai v2

问题描述

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

为此，我已导出为 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)

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

解决方法

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

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

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