src.clustering.affinity

Classes

Affinity

Methods:

Module Contents

class src.clustering.affinity.Affinity(dataset_name: str = None, embedding_method: str = None, dataset_id: int = None, embedding_id: int = None, embeddings: pandas.DataFrame = None, random_seed: int = 42)

Bases: clustering.BaseClustering

Methods:

fit(data: pd.DataFrame):

Fit the Affinity Propagation clustering algorithm to the data.

clusterer