如果重要的话,我会使用sklearn
,但我有:
regressor = RandomForestRegressor(
n_estimators=10, random_state=0, verbose=1, n_jobs=-1)
regressor.fit(X_train, np.ravel(y_train))
y_pred = regressor.predict(X_test)
我的y_pred
有shape
:(3000,)
我想将该列添加回X_test
,标头为job_performance
,最后保存为CSV。有人能帮上忙吗?
发布于 2019-06-26 04:06:27
然后保存为csv,如下所示
X_test['job_perf'] = y_pred
X_test.to_csv('filename.csv')
https://stackoverflow.com/questions/56761135
复制相似问题