cluster.py 文件源码

def find_best_cluster(cluster_type,data,a,b):
    #----Prints silhouette scores for all # of clusters in range
    scores = []
    for i in range(a,b):

        if cluster_type.lower() == "kmeans":
            i_clusters = kmeans(data, i)
        elif cluster_type.lower() == "agglom":
            i_clusters = agglom(data, i)

        sil_score_i = i_clusters['silhouette_score']
        scores.append(sil_score_i)

    print(scores)