Hitta viktiga prediktorer för hjärtsjukdom med SHAP
Din uppgift är att använda SHAP för att förstå hur olika särdrag i en förtränad RandomForestClassifier model påverkar prediktioner för hjärtsjukdom.
X med särdragen och y med etiketterna, samt den slumpmässiga skogklassificeraren model, har laddats in åt dig.
Den här övningen är en del av kursen
Förklarbar AI i Python
Övningsinstruktioner
- Skapa en SHAP tree explainer med namnet
explainer. - Beräkna
shap_values. - Beräkna det genomsnittliga absoluta SHAP-värdet
mean_abs_shap.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
import shap
# Create a SHAP Tree Explainer
explainer = ____
# Calculate SHAP values
shap_values = ____
# Calculate mean absolute SHAP values
mean_abs_shap = ____
plt.bar(X.columns, mean_abs_shap)
plt.title('Mean Absolute SHAP Values for RandomForest')
plt.xticks(rotation=45)
plt.show()