Comece agoraComece grátis

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 exercicio faz parte do curso

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

Ver curso

Instruções do exercicio

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

exercicio interativo prático

Tente este exercicio completando este código de exemplo.

# Fit gb to the training set
____

# Predict test set labels
y_pred = ____
Editar e Executar Código