Mãos à Obra: Aprendizado de Máquina com Scikit-Learn e TensorFlow

200 | Capítulo 7: Ensemble Learning e Florestas Aleatórias

Analisaremos um exemplo de regressão simples utilizando as Árvores de Decisão

como previsores base (claro que o Gradient Boosting também trabalha muito bem com

as tarefas de regressão). Isso é chamado Gradient Tree Boosting, ou Gradient Boosted

Regression Trees (GBRT). Primeiro, ajustaremos um DecisionTreeRegressor ao conjunto

de treinamento (por exemplo, um conjunto de treinamento quadrático ruidoso):

from sklearn.tree import DecisionTreeRegressor

tree_reg1 = DecisionTreeRegressor(max_depth=2)

tree_reg1.fit(X, y)

Agora treine um segundo DecisionTreeRegressor nos erros residuais cometidos ...