def KTR()

in orbit/models/ktr.py [0:0]

• 27 lines of code
• 1 McCabe index (conditional complexity)

def KTR(
    level_knot_scale=0.1,
    level_segments=10,
    level_knot_distance=None,
    level_knot_dates=None,
    # seasonality
    seasonality=None,
    seasonality_fs_order=None,
    seasonality_segments=3,
    seasonal_initial_knot_scale=1.0,
    seasonal_knot_scale=0.1,
    # regression
    regressor_col=None,
    regressor_sign=None,
    regressor_init_knot_loc=None,
    regressor_init_knot_scale=None,
    regressor_knot_scale=None,
    regression_segments=5,
    regression_knot_distance=None,
    regression_knot_dates=None,
    # different from seasonality
    regression_rho=0.15,
    # shared
    degree_of_freedom=30,
    date_freq=None,
    # time-based coefficient priors
    coef_prior_list=None,
    flat_multiplier=True,
    # TODO: rename to residuals upper bound
    residuals_scale_upper=None,
    ktrlite_optim_args=dict(),
    estimator="pyro-svi",
    **kwargs,