【发布时间】:2018-07-10 12:48:50
【问题描述】:
我尝试使用 Python API 中的 XGBRegressor 的 score 方法,它返回的结果为 0.917。我希望这是回归的 r2 分数。
但是,在同一个包上尝试来自 sklearn 的 r2_score,它返回一个不同的值 (0.903)
xgbr.score(x_test, y_test) # Returns 0.917
y_pred = xgbr.predict(x_test)
r2_score(y_pred, y_test) # Returns 0.903
发生了什么事?我找不到有关 XGBoost 评分方法的任何文档。我正在使用 v0.7
【问题讨论】:
标签: python scikit-learn xgboost