这个问题可能看起来很傻,但我没有得到一个想法,所以我需要你的帮助。
我使用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_pred
和y_test
数据的 Dataframe ,并将其保存为CSV格式。
你知道怎么做吗?
1条答案
按热度按时间cx6n0qe31#
看起来很简单,就在我的脑海里点击了一下。所以这样,就可以做到: