ComeçarComece gratuitamente

Treinar o regressor GB

Agora você treinará o regressor de gradient boosting gb que instanciou no exercício anterior e preverá os rótulos do conjunto de teste.

O conjunto de dados é dividido em 80% de treinamento e 20% de teste. As matrizes de recursos X_train e X_test, bem como as matrizes y_train e y_test estão disponíveis em seu espaço de trabalho. Além disso, também carregamos a instância do modelo gb que você definiu no exercício anterior.

Este exercício faz parte do curso

Aprendizado de máquina com modelos baseados em árvores em Python

Ver Curso

Instruções de exercício

  • Ajuste gb ao conjunto de treinamento.
  • Preveja os rótulos do conjunto de teste e atribua o resultado a y_pred.

Exercício interativo prático

Experimente este exercício preenchendo este código de exemplo.

# Fit gb to the training set
____

# Predict test set labels
y_pred = ____
Editar e executar código