在 python 中获取 model.predict 个值

Question

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

预测=model.predict(test_x)

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

数组([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

中完成

Answer 1

预测函数returns一个数组对象，因此您可以按如下方式将其转换为数据帧。

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'])

在 python 中获取 model.predict 个值

get model.predict values in python

python

prediction

output