将预测的随机森林结果转换为数据框答案

【问题标题】：Converting predicted random forest results into dataframe将预测的随机森林结果转换为数据框
【发布时间】：2021-10-09 09:07:22
【问题描述】：

这个问题可能看起来很傻，但没有得到任何想法，所以需要你的帮助。

我用random forest预测结果，写了如下代码

    from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=1)
from sklearn.ensemble import RandomForestRegressor
  
 # create regressor object
regressor = RandomForestRegressor(n_estimators = 100, random_state = 0)
  
# fit the regressor with x and y data
regressor.fit(X_train, y_train)
Y_pred = regressor.predict(X_test)

Y_pred 是给定X_test 的结果。现在，我想为我的Y_pred 和y_test 数据创建一个数据框并将其保存为CSV 格式。

知道怎么做吗？

【问题讨论】：

标签： python pandas regression random-forest

【解决方案1】：

这似乎很简单，只是点击进入我的脑海。这样就可以做到了

df_new = pd.DataFrame({'x':Y_pred, 'y':y_test}) df_new.head()

【讨论】：