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)

基类：StackingEnsembleBase

表征聚类

参数

• repr_model (ReprBasemodel) – 使用的表征模型

• repr_model_params (dict) – 表征模型的参数

• encode_params (dict) – 表征模型encode的参数

• downstream_learner (Callable) – 下游聚类器，需要是一个sklearn形式的聚类器，默认KMeans()

• verbose (bool) – 是否开启日志，默认开启

fit(train_datasets: List[TSDataset]) → None

参数

tsdataset_list (TSDataset) – 训练集

predict(tsdatasets: List[TSDataset]) → ndarray

预测

参数

tsdataset_list (TSDataset) – 预测数据

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

保存模型

参数

• path (str) – 保存路径

• ensemble_file_name (str) – 保存文件名

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

加载模型

参数

• path (str) – 加载路径

• ensemble_file_name (str) – 保存文件名

返回

加载的模型