K-means for feedback clustering
You have a dataset of feedback responses, and you've used a GPT model to calculate confidence scores for each response. To identify unusual or outlier feedback, you apply k-means clustering to the low-confidence responses.
The KMeans algorithm, reviews and confidences variables, and np library have been preloaded.
This exercise is part of the course
Reinforcement Learning from Human Feedback (RLHF)
Exercise instructions
- Initialize the k-means algorithm. Set the
random_stateto42for code reproducibility. - Calculate distances from cluster centers to identify outliers as the difference between
dataand the corresponding cluster centers.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
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)