Clustering

class Clustering[source]

Bases: abc.ABC, etna.core.mixins.BaseMixin

Base class for ETNA clustering algorithms.

Inherited-members

Methods

fit_predict()

Fit clustering algo and predict clusters.

get_centroids()

Get centroids of clusters.

to_dict()

Collect all information about etna object in dict.
abstract fit_predict() → Dict[str, int][source]

Fit clustering algo and predict clusters.

Returns

dict in format {segment: cluster}

Return type

Dict[str, int]

abstract get_centroids() → pandas.core.frame.DataFrame[source]

Get centroids of clusters.

Returns

dataframe with centroids

Return type

pd.DataFrame