CommencezCommencez gratuitement

K-means pour regrouper les rétroactions

Vous disposez d'un jeu de données de rétroactions, et vous avez utilisé un modèle GPT pour calculer un score de confiance pour chaque réponse. Pour repérer les rétroactions inhabituelles ou aberrantes, vous appliquez le regroupement k-means aux réponses ayant une faible confiance.

L'algorithme KMeans, les variables reviews et confidences, ainsi que la bibliothèque np ont été préchargés.

Cette activité fait partie du cours

Reinforcement Learning from Human Feedback (RLHF)

Voir le cours

Instructions de l’exercice

  • Initialisez l'algorithme k-means. Réglez random_state à 42 pour assurer la reproductibilité du code.
  • Calculez les distances depuis les centres des regroupements pour identifier les valeurs aberrantes en tant que différence entre data et les centres de regroupement correspondants.

Exercice interactif pratique

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

def detect_anomalies(data, n_clusters=3):
    # Initialize k-means
    ____
    clusters = kmeans.fit_predict(data)
    centers = kmeans.cluster_centers_

    # Calculate distances from cluster centers
    ____
    return distances
  
anomalies = detect_anomalies(confidences)
print(anomalies)
Modifier et exécuter le code