CommencezCommencez gratuitement

Importance des caractéristiques en regroupement avec l'ARI

Exploitez l'Adjusted Rand Index (ARI) pour mesurer quantitativement l'effet du retrait de chaque caractéristique sur l'attribution des grappes dans le jeu de données clients utilisé à l'exercice précédent, déjà chargé dans X.

La fonction adjusted_rand_score() et la variable column_names ont été préchargées pour vous.

Cette activité fait partie du cours

Explainable AI en Python

Voir le cours

Instructions de l’exercice

  • Calculez les attributions de grappes d'origine dans original_clusters.
  • Dans la boucle for, retirez les caractéristiques une à une et enregistrez le résultat dans X_reduced.
  • Obtenez reduced_clusters en appliquant K-means sur X_reduced.
  • Calculez l'importance de chaque caractéristique à partir de l'ARI entre reduced_clusters et original_clusters.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

kmeans = KMeans(n_clusters=5, random_state=10, n_init=10).fit(X)
# Derive original clusters
original_clusters = ____

for i in range(X.shape[1]):
  	# Remove feature at index i
    X_reduced = ____
    # Derive reduced clusters
    reduced_clusters = ____
    # Derive feature importance
    importance = ____
    print(f'{column_names[i]}: {importance}')
Modifier et exécuter le code