添加多项式特征后如何预测测试数据？

Question

我正在使用房价数据集创建多项式特征。但是在将这些功能添加到我的训练集中后，X_train不再适合X_test。

X火车形状：（17290，18） 添加新功能后， X_train形状：（17290，22）

但是我的X_test形状仍然是（4323，18）。预测测试值时出现错误。那么添加一些多项式特征后该怎么办？

Answer 1

您的模型已更改，在将特征添加到预测数组之前，您似乎仍在插入相同数量的特征。

myarraytest = np.array(sampledata)
testdata = myarraytest.reshape(1,len(myarraytest))
reg.predict(testdata) #your model name

您应该将4个新功能添加到您要预测的数组中，以使形状相同。