微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何获得一列文本 fastai v2

如何解决如何获得一列文本 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 举报,一经查实,本站将立刻删除。