kmeans_cluster_ioc.py

from sklearn.cluster import KmeansClustering

from cluster_template import ClusterTemplate

class KmeansCluster(ClusterTemplate):
    
    def __init__(self, path):
        super(KmeansCluster, self).__init__(path)
        
    # Implementation of abstract method
    def init_cluster_algo(self, num_clusters):
        self.clustering_algo = KmeansClustering(nclusters=num_clusters)