K-means для кластеризації відгуків
У вас є набір даних із відповідями на відгуки, і ви використали модель GPT, щоб обчислити оцінки впевненості для кожної відповіді. Щоб знайти незвичні або «викидні» відгуки, ви застосовуєте кластеризацію k-means до відповідей із низькою впевненістю.
Алгоритм KMeans, змінні reviews і confidences, а також бібліотека np вже завантажені.
Ця вправа є частиною курсу
Reinforcement Learning from Human Feedback (RLHF)
Інструкції до вправи
- Ініціалізуйте алгоритм k-means. Встановіть
random_stateна42для відтворюваності коду. - Обчисліть відстані від центрів кластерів, щоб визначити викиди як різницю між
dataта відповідними центрами кластерів.
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
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)