将任何sklearn结果映射回原始数据帧

时间:2016-12-19 08:51:05

标签: python pandas scikit-learn sklearn-pandas

我想用原始测试数据作为参考来分析我的随机森林结果的预测值。

预测结果出现在数组中,因为我使用它:

predict = rf.predict(test[columns])

如何使用pandas将预测结果映射回原始数据集?

此致 galeej

1 个答案:

答案 0 :(得分:-1)

EdChum回答了这个问题。

(编辑)他的回答:

  

您应该只需将其添加为新列:   测试['预测'] = rf.predict(测试[列])

我对延迟表示歉意。

此致 苏尼