ÎncepețiÎncepe gratuit

Identificarea principalilor predictori ai bolilor de inimă cu SHAP

Sarcina ta este să folosești SHAP pentru a înțelege cum influențează diferitele caracteristici ale unui model RandomForestClassifier pre-antrenat predicțiile pentru bolile de inimă.

X cu caracteristicile, y cu etichetele și clasificatorul random forest model sunt deja încărcate.

Acest exercițiu face parte din cursul

AI Explicabil în Python

Vezi cursul

Instrucțiuni pentru exercițiu

  • Creează un explainer SHAP de tip tree, numit explainer.
  • Calculează shap_values.
  • Calculează media valorilor absolute SHAP, mean_abs_shap.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

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()
Editează și rulează codul