ComenzarEmpieza gratis

Importancia de las variables

Aunque algunos atributos de los caramelos, como el chocolate, sean muy populares, eso no significa que vayan a ser importantes para la predicción del modelo. Una vez que hayas ajustado un modelo de random forest, puedes revisar el atributo del modelo .feature_importances_ para ver qué variables tuvieron mayor impacto. Puedes comprobar la importancia de cada variable en el modelo iterando sobre el array de importancias con enumerate().

Si no conoces la función enumerate() de Python, te permite recorrer una lista mientras crea un contador automático.

Este ejercicio forma parte del curso

Validación de modelos en Python

Ver curso

Instrucciones del ejercicio

  • Recorre la salida de importancias de rfr.
  • Imprime los nombres de las columnas de X_train y la puntuación de importancia de cada columna.

Ejercicio interactivo práctico

Prueba este ejercicio y completa el código de muestra.

# Fit the model using X and y
rfr.fit(X_train, y_train)

# Print how important each column is to the model
for i, item in enumerate(rfr.____):
      # Use i and item to print out the feature importance of each column
    print("{0:s}: {1:.2f}".format(X_train.columns[____], ____))
Editar y ejecutar código