ComenzarEmpieza gratis

Entrena tu primer árbol de regresión

En este ejercicio, entrenarás un árbol de regresión para predecir el consumo de mpg (millas por galón) de los coches del conjunto de datos auto-mpg utilizando las seis características disponibles.

El conjunto de datos se procesa por ti y se divide en un 80 % de entrenamiento y un 20 % de prueba. La matriz de características X_train y la matriz y_train están disponibles en tu espacio de trabajo.

Este ejercicio forma parte del curso

Machine learning con modelos basados en árboles en Python

Ver curso

Instrucciones de ejercicio

  • Importa DecisionTreeRegressor desde sklearn.tree.
  • Instancia un DecisionTreeRegressor dt con profundidad máxima 8 y min_samples_leaf fijado en 0,13.
  • Ajusta dt al conjunto de entrenamiento.

Ejercicio interactivo práctico

Pruebe este ejercicio completando este código de muestra.

# Import DecisionTreeRegressor from sklearn.tree
from ____.____ import ____

# Instantiate dt
____ = ____(max_depth=____,
             ____=____,
            random_state=3)

# Fit dt to the training set
____.____(____, ____)
Editar y ejecutar código