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)
Instructions de l’exercice
- Initialisez l'algorithme k-means. Réglez
random_stateà42pour assurer la reproductibilité du code. - Calculez les distances depuis les centres des regroupements pour identifier les valeurs aberrantes en tant que différence entre
dataet 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)