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
Oefeninstructies
- Haal de feature-importance uit het
model. - Plot de
feature_importancesvoor de gegevenfeature_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()