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
Instrucciones de ejercicio
- Importa
DecisionTreeRegressor
desdesklearn.tree
. - Instancia un
DecisionTreeRegressor
dt
con profundidad máxima 8 ymin_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
____.____(____, ____)