Source code for lightwood.mixer.prophet

from typing import Dict, Union

from lightwood.mixer.sktime import SkTime


[docs]class ProphetMixer(SkTime): def __init__(self, stop_after: float, target: str, dtype_dict: Dict[str, str], horizon: int, ts_analysis: Dict, model_path: str = 'fbprophet.Prophet', auto_size: bool = True, sp: int = None, hyperparam_search: bool = False, use_decomposers: Dict[str, Union[int, str]] = {} ): super().__init__(stop_after, target, dtype_dict, horizon, ts_analysis, model_path, auto_size, sp, hyperparam_search, use_decomposers) self.stable = False self.model_path = model_path self.possible_models = [self.model_path] self.n_trials = len(self.possible_models)