cluster.py

import os

from abc import abstractmethod

class Cluster(object):

    def __init__(self, path):
        self.path = path
        if not os.path.exists(self.path):
            os.makedirs(self.path, exist_ok=True)

    @abstractmethod
    def cluster(self, num_centroids):
        raise NotImplementedError("Abstract method 'cluster' not implemented")