sklearn笔记21 梯度提升树的使用

完整代码 sklearn代码15 5-梯度提升树

参数使用：



官网信息

import numpy as np

# f(x) = 3*x**2
# 梯度=导数
from sklearn.ensemble import GradientBoostingClassifier

from sklearn import datasets

from sklearn.model_selection import train_test_split

X,y = datasets.load_iris(True)

X_train,X_test, y_train,y_test = train_test_split(X,y,test_size = 0.2)

gbdt = GradientBoostingClassifier(n_estimators=10)

gbdt.fit(X_train,y_train)

gbdt.score(X_test, y_test)

gbdt.n_estimators_

gbdt.estimators_    #返回十颗树

y

gbdt.estimators_.shape

三类划分：

其余不变

X,y = datasets.load_iris(True)

cond = y!=2
X = X[cond]
y = y[cond]

X_train,X_test, y_train,y_test = train_test_split(X,y,test_size = 0.2)