开始使用免费开始使用

用于反馈聚类的 K-means

您有一个反馈回复的数据集,并使用 GPT 模型为每条回复计算了置信度分数。为识别异常或离群的反馈,您将对低置信度的回复应用 k-means 聚类。

KMeans 算法、reviewsconfidences 变量,以及 np 库都已预加载。

本练习是课程的一部分

来自人类反馈的强化学习(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)
编辑并运行代码