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
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 = ____