将任何 sklearn 结果映射回原始数据帧
mapping back any sklearn result to the original dataframe
我想以原始测试数据为参考,分析我的随机森林结果在excel中的预测值。
当我使用这个时,预测结果出现在一个数组中:
predict = rf.predict(test[columns])
如何使用 pandas 将预测结果映射回原始数据集?
此致,
galeej
EdChum 回答了问题。
(编辑)他的回答:
You should be able to just add this back as a new column:
test['prediction'] = rf.predict(test[columns])
对于延迟,我深表歉意。
此致,
苏尼尔
我想以原始测试数据为参考,分析我的随机森林结果在excel中的预测值。
当我使用这个时,预测结果出现在一个数组中:
predict = rf.predict(test[columns])
如何使用 pandas 将预测结果映射回原始数据集?
此致, galeej
EdChum 回答了问题。
(编辑)他的回答:
You should be able to just add this back as a new column: test['prediction'] = rf.predict(test[columns])
对于延迟,我深表歉意。
此致, 苏尼尔