ComeçarComece gratuitamente

Instanciar o modelo

No conjunto de exercícios a seguir, você diagnosticará os problemas de viés e variância de uma árvore de regressão. A árvore de regressão que você definirá neste exercício será usada para prever o consumo de mpg dos carros do conjunto de dados de automóveis usando todos os recursos disponíveis.

Já processamos os dados e carregamos a matriz de recursos X e a matriz y em seu espaço de trabalho. Além disso, a classe DecisionTreeRegressor foi importada de sklearn.tree.

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

  • Importe train_test_split de sklearn.model_selection.
  • Divida os dados em 70% de treinamento e 30% de teste.
  • Instancie um DecisionTreeRegressor com profundidade máxima de 4 e min_samples_leaf definido como 0,26.

Exercício interativo prático

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

# Import train_test_split from sklearn.model_selection
____

# Set SEED for reproducibility
SEED = 1

# Split the data into 70% train and 30% test
X_train, X_test, y_train, y_test = ____(____, ____, test_size=____, random_state=SEED)

# Instantiate a DecisionTreeRegressor dt
dt = ____(____=____, ____=____, random_state=SEED)
Editar e executar código