Aan de slagGa gratis aan de slag

Finding key heart disease predictors with SHAP

Your task is to use SHAP to understand how different features in a pre-trained RandomForestClassifier model influence predictions for heart disease.

X containing the features and y containing the labels, and the random forest classifier model have been pre-loaded for you.

Deze oefening maakt deel uit van de cursus

Explainable AI in Python

Cursus bekijken

Oefeninstructies

  • Create a SHAP tree explainer named explainer.
  • Compute shap_values.
  • Compute the mean absolute SHAP values mean_abs_shap.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

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()
Code bewerken en uitvoeren