src.clustering.kmeans

Classes

KMeansClustering

Methods:

Module Contents

class src.clustering.kmeans.KMeansClustering(dataset_name: str = None, embedding_method: str = None, dataset_id: int = None, embedding_id: int = None, embeddings: pandas.DataFrame = None, n_clusters: int = 8, random_seed: int = 42)

Bases: clustering.BaseClustering

Methods:

fit(data: pd.DataFrame):

Fit the KMeans clustering algorithm to the data.

clusterer