1. Apprendre
    2. /
  2. Cours
    3. /
  3. Machine Learning avec des modèles à base d'arbres en Python
Connected

Exercice

Visualiser l'importance des caractéristiques

Dans cet exercice, vous allez déterminer quelles caractéristiques ont été les plus prédictives selon le régleur de forêts aléatoires rf que vous avez entraîné dans un exercice précédent.

Pour ce faire, vous allez tracer un diagramme à bandes horizontal représentant l'importance des caractéristiques telle qu'évaluée par rf. Heureusement, cela se fait facilement grâce aux fonctionnalités de traçage de pandas.

Nous avons créé un objet pandas.Series appelé importances qui contient les noms des caractéristiques comme index et leurs importances comme valeurs. De plus, matplotlib.pyplot est disponible sous le nom plt et pandas sous pd.

Instructions

100 XP

  • Appelez la méthode .sort_values() sur importances et assignez le résultat à importances_sorted.

  • Appelez la méthode .plot() sur importances_sorted et définissez les arguments :

    • kind à 'barh'
    • color à 'lightgreen'