from sklearn.cluster import KMeans


class KMeansClustering:

    def __init__(self):
        pass

    def perform_clustering(self, feature_vectors, num_clusters):

        kmeans = KMeans(n_clusters=num_clusters).fit(feature_vectors)
        labels = kmeans.labels_

        return labels
