sklearn中的gbdt源码分析

简单看了一下sklearn中的gbdt源码
在这里记录一下,目前来讲还不能把gbdt的全部理论都搞清楚
sklearn有两个相似于gbdt的实现函数

GradientBoostingClassifier
GradientBoostingRegressor

一个用于分类,另外一个用于回归
这两个类其实区别只在于mixin上,下面是类继承关系code

GradientBoostingRegressor继承了
    BaseGradientBoosting
    RegressorMixin
GradientBoostingClassifier继承了
    BaseGradientBoosting
    ClassifierMixin
ClassifierMixin和RgressorMixin的区别:
    classifierMixin使用的是准确率来计算偏差
    而回归的是使用r_square来计算偏差
实际上这两个模型的差距是很小的,就是计算一下拟合度

而后是整个训练的过程
训练的过程会先设置一些参数
设置n_estimators是要训练的分类器数据
若是损失函数是loss的话,那么就比较简单了
每次训练弱分类器都会根据上一次的结果来
上次生成的y - y_pred会做为新的y传进去
这样来训练直到n_estimators足够继承

不过在predict结果的时候有点看不懂代码
后面看了再补充源码

相关文章
相关标签/搜索