Get startedGet started for free

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)

View Course

Exercise instructions

  • Initialize the k-means algorithm. Set the random_state to 42 for code reproducibility.
  • Calculate distances from cluster centers to identify outliers as the difference between data and 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)
Edit and Run Code