ComenzarEmpieza gratis

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

Ver curso

Instrucciones del ejercicio

  • Añade un parámetro a rfr para 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 con X_train como datos de entrada y y_train como 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.____(____, ____)
Editar y ejecutar código