如何将梯度提升回归结果保存到文件中？

Question

我有一个梯度提升回归模型，我想将其结果保存到 csv。

from sklearn import ensemble
clf = ensemble.GradientBoostingRegressor(n_estimators = 400, max_depth = 5, min_samples_split = 2,
          learning_rate = 0.1, loss = 'squared_error')
clf.fit(x_train, y_train)

我想查看实际值和预测值。

Answer 1

在拟合你的模型之后，你实际上需要predict 预测，clf.predict，假设我们要保存y_pred_train & y_pred_test：

y_pred_train = clf.predict(y_train)
y_pred_test = clf.predict(y_test)

现在我们已经有了train 和test 集合的预测值和实际值，我们可以使用np.savetxt 保存.csv 文件：

np.savetxt("train_preds.csv", np.column_stack((y_train, y_pred_train)), delimiter=",", fmt='%.5f')

和

np.savetxt("test_preds.csv", np.column_stack((y_test, y_pred_test)), delimiter=",", fmt='%.5f')

我们使用 np.column_stack 只是将它们堆叠为列而不是行，因为它们本质上是一维 numpy 数组。

请注意，我格式化浮点精度只是为了便于查看，如果您要在之后计算损失，最好不要格式化。