XGBoost回归预测

Question

我训练了一个XGBoost回归模型，它试图预测运动提供的转换数。自变量是每月虚拟，位置虚拟，和4列竞选规则(数字)。总共有6列ind。变量。我训练了模特。现在我试着用这个模型来预测一个特定的竞选活动的表现。我的目标是输入新活动的规则、月份和地点，并得到XGBoost预测的转换量。您知道如何在Python中实现这种预测吗？谢谢

Answer 1

您应该能够简单地使用predict方法，并给它与您试图得到的预测值相同的值，并且模型将输出预测值。只需确保要预测的数据与用于培训的数据具有相同的格式(即，相同的列数和值的缩放(如果适用)，并将其作为预测方法的X参数传递。如果您只想对一个观察结果进行预测，您可能需要添加一个额外的维度，这样您就有了一个形状(1, 5)数组(或者您有多少列)，因为模型很可能需要一个2D数组。