Source code for dsframework.interfaces.experiments

from abc import ABC, abstractmethod


class BaseExperiment(ABC):  # pragma: no cover
    """Base class for all experiments.
    """

    @abstractmethod
    def get_data(self, datetime_column: str):
        pass

    @abstractmethod
    def prepare_data(self, *args, **kwargs):
        pass

    def init_fit(self, *args, **kwargs):
        pass

    @abstractmethod
    def predict(self, *args, **kwargs):
        pass

    def calc_metrics(self, *args, **kwargs):
        pass