aiq.profiler.forecasting.models.forecasting_base_model#

Classes#

ForecastingBaseModel

Abstract base class for all models in this package.

Module Contents#

class ForecastingBaseModel#

Bases: abc.ABC

Abstract base class for all models in this package.

abstractmethod fit(raw_stats)#

Train/fine-tune the model on the provided dataset.

abstractmethod predict(raw_stats) numpy.ndarray#

Predict using the trained model. Returns a np.ndarray, shape = (N, 4).