Aan de slagGa gratis aan de slag

Feature-importance berekenen met beslisbomen

Je hebt een decision tree-classifier gebouwd om patiënten met risico op hartziekte te identificeren met de heart disease-gegevensset. Nu moet je het model uitleggen door de feature-importance te analyseren om de belangrijkste factoren voor het voorspellen van hartziekte te bepalen, zodat gerichtere zorginterventies mogelijk worden.

matplotlib.pyplot is geïmporteerd als plt. X_train en y_train zijn al voor je geladen.

Deze oefening maakt deel uit van de cursus

Explainable AI in Python

Cursus bekijken

Oefeninstructies

  • Haal de feature-importance uit het model.
  • Plot de feature_importances voor de gegeven feature_names.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

model = DecisionTreeClassifier(random_state=42)
model.fit(X_train, y_train)

# Derive feature importances
feature_importances = ____
feature_names = X_train.columns

# Plot the feature importances
____
plt.show()
Code bewerken en uitvoeren