在 python 中获取 model.predict 值答案

【问题标题】：get model.predict values in python在 python 中获取 model.predict 值
【发布时间】：2018-04-23 07:26:48
【问题描述】：

我正在运行一个随机森林模型。我正在使用以下代码预测值

预测 = model.predict(test_x)

我想要来自下面输出的列表或数据框中的预测值。

array([1, 1, 1, 1, 0, 0, 0, 1, 0, 0, 0, 0, 1, 1, 0, 0, 1, 1, 1, 0, 1, 1, 1、 0, 0, 1, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 1, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0, 1, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 1, 1, 1, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 1, 0, 0, 1, 0, 1, 0], dtype=int64)

如何在 Python 中实现

【问题讨论】：

你用的是什么模块？ model 并没有告诉我们太多。
从 sklearn.ensemble 导入 RandomForestClassifier 作为 rf
你检查过documentation吗？您实际尝试过什么？

标签： python output prediction

【解决方案1】：

predict 函数返回一个数组对象，因此您可以将其转换为数据帧，如下所示。

import pandas as pd 
prediction = model.predict(test_x)

cols = prediction[0].keys()
df = pd.DataFrame([[getattr(i,j) for j in cols] for i in prediction], columns = cols)

针对您的特殊情况：

import pandas as pd 
prediction = model.predict(test_x)
df = pd.DataFrame(prediction, columns = ['ouput'])

【讨论】：

它给出错误 - 'numpy.int64' 对象没有属性 'keys'
乐于助人:)