Configura parámetros y ajusta un modelo
Las tareas predictivas se dividen en dos categorías: regresión o clasificación. En el conjunto de datos de caramelos, el resultado es una variable continua que describe con qué frecuencia se eligió un caramelo frente a otro en una serie de duelos 1 contra 1. Para predecir este valor (el porcentaje de victorias), usarás un modelo de regresión.
En este ejercicio, especificarás algunos parámetros usando un modelo de regresión de bosque aleatorio rfr.
Este ejercicio forma parte del curso
Validación de modelos en Python
Instrucciones del ejercicio
- Añade un parámetro a
rfrpara que el número de árboles sea 100 y la profundidad máxima de esos árboles sea 6. - Asegúrate de que el modelo sea reproducible añadiendo un estado aleatorio de
1111. - Usa el método
.fit()para entrenar el modelo de regresión de bosque aleatorio conX_traincomo datos de entrada yy_traincomo respuesta.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# Set the number of trees
rfr.____ = ____
# Add a maximum depth
rfr.____ = ____
# Set the random state
rfr.____ = ____
# Fit the model
rfr.____(____, ____)