discontinuum.engines.base

discontinuum.engines.base#

Functions

is_fitted(func)

Decorator checks whether model has been fit.

Classes

BaseModel([model_config])

class discontinuum.engines.base.BaseModel(model_config: Dict | None = None)#
abstractmethod build_datamanager()#

Build DataManager for the model.

abstractmethod build_model(X, y)#

TODO Sometimes this sets self.model and sometimes this returns model. Not sure that we can standardize the behavior for different engines.

abstractmethod fit(covariates: Dataset, target: Dataset, **kwargs)#

Fit model to data.

Parameters:

  • covariates (Dataset) – Covariates for training.

  • target (Dataset) – Target data for training.

  • kwargs (dict) – Additional keyword arguments.

abstractmethod predict(covariates: Dataset) → DataArray#

Use a fitted model to make predictions on new data.

Parameters:

covariates (Dataset) – Covariates for prediction.

discontinuum.engines.base.is_fitted(func)#

Decorator checks whether model has been fit.

Contents