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)