paddlets.models.representation.task.repr_cluster

class ReprCluster(repr_model: ReprBaseModel, repr_model_params: Optional[dict] = None, encode_params: Optional[dict] = None, downstream_learner: Optional[Callable] = None, verbose: bool = False)[source]

Bases: StackingEnsembleBase

The ReprCluster Class.

Parameters

  • repr_model (ReprBasemodel) – Representation model to use for cluster.

  • repr_model_params (dict) – params for reprmodel init.

  • encode_params (dict) – params for reprmodel encode.

  • downstream_learner (Callable) – The downstream learner, should be a sklearn-like cluster, set to KMeans() by default.

  • verbose (bool) – Turn on Verbose mode,set to true by default.

fit(train_datasets: List[TSDataset]) → None[source]
Parameters

tsdataset_list (TSDataset) – train data.

predict(tsdatasets: List[TSDataset]) → ndarray[source]

Predict

Parameters

tsdataset_list (TSDataset) – predict data.

save(path: str, repr_cluster_file_name: str = 'repr-cluster-partial.pkl') → None[source]

Save the repr-cluster model to a directory.

Parameters

  • path (str) – Output directory path.

  • ensemble_file_name (str) – Name of repr-cluster model object. This file contains meta information of repr-cluster model.

static load(path: str, repr_cluster_file_name: str = 'repr-cluster-partial.pkl') → ReprCluster[source]

Load the repr-cluster model from a directory.

Parameters

  • path (str) – Input directory path.

  • ensemble_file_name (str) – Name of repr-cluster model object. This file contains meta information of repr-cluster model.

Returns

The loaded ensemble model.