paddlets.ensemble.base

class EnsembleBase(estimators: Optional[List[Tuple[object, dict]]] = None, verbose: bool = False)

基类：object

The EnsembleBase Class.

参数

• estimators (List[Tuple[object, dict]]) – 一个由paddlets模型组成的 tuple (模型类名,模型参数) 列表,]

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

abstract fit(train_tsdataset: TSDataset, valid_tsdataset: Optional[TSDataset] = None) → None

Fit

参数

• train_tsdataset (TSDataset) – 训练集

• valid_tsdataset (TSDataset, optional) – 验证集

abstract predict(tsdataset: TSDataset) → None

Predict

参数

tsdataset (TSDataset) – 预测数据

save(path: str, ensemble_file_name: str = 'paddlets-ensemble-partial.pkl') → None

保存模型

参数

• path (str) – 保存路径

• ensemble_file_name (str) – 模型文件名

static load(path: str, ensemble_file_name: str = 'paddlets-ensemble-partial.pkl') → EnsembleBase

加载模型

参数

• path (str) – 加载路径

• ensemble_file_name (str) – 模型文件名

返回

加载的模型