paddlets.ensemble.base

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

Bases: object

The EnsembleBase Class.

Parameters

  • estimators (List[Tuple[object, dict]]) – A list of tuple (class,params) consisting of several paddlets models.

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

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

Fit

Parameters

  • train_tsdataset (TSDataset) – Train dataset.

  • valid_tsdataset (TSDataset, optional) – Valid dataset.

abstract predict(tsdataset: TSDataset) → None[source]

Predict

Parameters

tsdataset (TSDataset) – Dataset to predict.

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

Save the ensemble model to a directory.

Parameters

  • path (str) – Output directory path.

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

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

Load the ensemble model from a directory.

Parameters

  • path (str) – Input directory path.

  • ensemble_file_name (str) – Name of ensemble object. This file contains meta information of ensemble.

Returns

The loaded ensemble model.