首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >XGBRegressor与XGBClassifier的区别

XGBRegressor与XGBClassifier的区别
EN

Data Science用户
提问于 2018-04-05 22:50:36
回答 1查看 10.4K关注 0票数 5

我试图理解xgboost.XGBRegressor和xgboost.sklearn.XGBClassifier之间的区别。

有人能用简洁的方式解释这种差异吗?

因为当我用完全相同的数据对两个分类器进行匹配时,我得到了非常不同的性能。

我就是这么拟合数据的。

代码语言:javascript
复制
clf = xgboost.XGBRegressor(alpha=c)
#clf = xgboost.sklearn.XGBClassifier(alpha=c)

clf.fit(X_train, y_train)
y_pred = clf.predict(X_test)
print('Model roc auc score: %0.3f' % roc_auc_score(y_test, y_pred, average='macro', sample_weight=None))
print('Model pr auc score: %0.3f' % average_precision_score(y_test, y_pred))

当clf = xgboost.sklearn.XGBClassifier(alpha=c)时

模型中华大学成绩: 0.544 模型pr评分: 0.303

时clf = xgboost.XGBRegressor(alpha=c)

模型中华大学成绩: 0.703

模型pr评分: 0.453

是什么导致了这种性能差异?

EN

回答 1

Data Science用户

发布于 2018-04-05 23:06:21

XGBRegressor用于连续的目标/结果变量。这些常被称为“回归问题”。

XGBClassifier用于分类目标/结果变量。这些常被称为“分类问题”。

票数 13
EN
页面原文内容由Data Science提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://datascience.stackexchange.com/questions/29969

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档