pytorch - 在表格数据上创建 fastai 模型后如何对新数据进行评分？

Question

创建一个学习者

learn.fit_one_cycle(5)

我可以从下面的测试集代码中检查预测

preds_test, _ = learn.get_preds(ds_type=DatasetType.Test)
pred_test_prob, pred_test_class = preds_test.max(1)

如何检查不是数据集的新数据集的预测？我在下面尝试

validateObj = TabularList.from_df(validate, cat_names=cat_names, cont_names=cont_names, procs=procs)

preds, _ = learn.get_preds(validateObj)
pred_prob, pred_class = preds.max(1)

但它给了我一个与验证集108000比较的数组。45000

有没有一种简单的方法来使用上述学习器对新数据进行评分？

标签： pytorchtabularfast-ai