如何处理 Pycaret 在建模时添加额外功能？(用于重用模型)

Question

导入Pycaret后，我调用setup(mydf, 'mytarget')并运行compare_models()。然后，我想从比较列表中保存一个模型，并在另一个数据集上使用它。我所做的是：lr = create_model('lr')。

然而，当我尝试lr.predict(mynewdfwithouttarget)时，我得到了大小不匹配的错误：

X每个样本有11个特征；预计37

列表中的其他模型也输出相同(或类似(的错误。

那么，如何使用在compare_models()中训练的模型呢？

谢谢。

Answer 1

创建模型：

lr = create_model('lr')

预测测试/保持样本：

predict_model(lr);

最终确定部署模型：

final_lr = finalize_model(lr)

预测新数据：

predictions = predict_model(final_lr, data = mynewdfwithouttarget)