pandas 将预测的随机森林结果转换为 Dataframe

j8ag8udp  于 2023-03-28  发布在  其他
关注(0)|答案(1)|浏览(148)

这个问题可能看起来很傻,但我没有得到一个想法,所以我需要你的帮助。
我使用RandomForestRegressor()来预测结果,并编写了以下代码:

from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestRegressor

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=1)

      
# create a 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_predy_test数据的 Dataframe ,并将其保存为CSV格式。
你知道怎么做吗?

cx6n0qe3

cx6n0qe31#

看起来很简单,就在我的脑海里点击了一下。所以这样,就可以做到:

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

相关问题