码迷,mamicode.com
首页 > 其他好文 > 详细

XGBoost小记

时间:2019-02-01 23:50:56      阅读:296      评论:0      收藏:0      [点我收藏+]

标签:model   pre   should   number   min   base   mat   you   val   

1.原理

//TODO

2.Python Package Scikit-Learn API

2.1输入

数据的特征分为两类,一类是连续型,比如:体重,一种是分类型,比如性别。

在scikit-learn中的Glossary of Common Terms and API Elements有这么一段话:

Categorical Feature

A categorical or nominal feature is one that has a finite set of discrete values across the population of data. These are commonly represented as columns of integers or strings. Strings will be rejected by most scikit-learn estimators, and integers will be treated as ordinal or count-valued. For the use with most estimators, categorical variables should be one-hot encoded. Notable exceptions include tree-based models such as random forests and gradient boosting models that often work better and faster with integer-coded categorical variables. OrdinalEncoder helps encoding string-valued categorical features as ordinal integers, and OneHotEncoder can be used to one-hot encode categorical features. See also Encoding categorical features and the http://contrib.scikit-learn.org/categorical-encoding package for tools related to encoding categorical features.

大意是在利用基于树的模型训练时推荐使用数值编码而不是one-hot编码。

详情:https://scikit-learn.org/stable/glossary.html#glossary

2.2输出

在这里只说两点:multi:softmax和multi:softprob,官方文档是这么说的:

multi:softmax: set XGBoost to do multiclass classification using the softmax objective, you also need to set num_class(number of classes)
multi:softprob: same as softmax, but output a vector of ndata * nclass, which can be further reshaped to ndata * nclass matrix. The result contains predicted probability of each data point belonging to each class.

在这里略坑,建立model时无论填哪一个,在model fit之后,打印模型时参数却都是multi:softprob,但是predict的结果也和上述解释也不一致,结果是multi:softmax的结果,只有预测的标签,没有概率分布。

官方代码如下:可见num_class也是不用设置的,objective被强制替换成了multi:softprob.最后若想输出概率分布请用predict_proba函数来预测.

self.classes_ = np.unique(y)
self.n_classes_ = len(self.classes_)

if self.n_classes_ > 2:
            # Switch to using a multiclass objective in the underlying XGB instance
            xgb_options["objective"] = "multi:softprob"
            xgb_options[num_class] = self.n_classes_

3.DEMO

//TODO

XGBoost小记

标签:model   pre   should   number   min   base   mat   you   val   

原文地址:https://www.cnblogs.com/0xcafe/p/10347304.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!