【发布时间】:2019-07-31 12:02:37
【问题描述】:
在 XGBoost Regression 中预测价格,如何获得模型的系数、截距?如何像我们在线性回归的 Statsmodel 中获得模型的摘要? 见下面代码
from xgboost import XGBRegressor
# fit model no training data
model = XGBRegressor()
model.fit(X_train, y_train)
# make predictions for test data
y_pred = model.predict(X_test)
print("R^2: {}".format(model.score(X_test, y_test)))
rmse = np.sqrt(mean_squared_error(y_test, y_pred))
print("Root Mean Squared Error: {}".format(rmse))
这就是我构建模型并尝试获得如下系数的方式:
#print the intercept
print(model.intercept_)
AttributeError: Intercept (bias) is not defined for Booster type gbtree
print(model.coef_)
AttributeError: Coefficients are not defined for Booster type gbtree
谁能帮我解决这个问题。谢谢。
【问题讨论】:
标签: regression xgboost