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