def create_model_grid()

in afa/core.py [0:0]

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

def create_model_grid():
    """Make the "grid" of model configurations to explore.

    Returns
    -------
    list

    """
    grid = [
        ("naive", partial(naive)),
        ("naive|local", partial(naive, local_model=True)),
        ("naive|seasonal", partial(naive, seasonal=True)),
        ("naive|local|seasonal",
            partial(naive, local_model=True, seasonal=True)),

        ("naive|log", partial(naive, use_log=True)),
        ("naive|log|local", partial(naive, use_log=True, local_model=True)),
        ("naive|log|seasonal", partial(naive, use_log=True, seasonal=True)),
        ("naive|log|local|seasonal",
            partial(naive, use_log=True, local_model=True, seasonal=True)),

        ("trend", partial(trend)),
        ("trend|local", partial(trend, local_model=True)),
        ("trend|seasonal", partial(trend, seasonal=True)),
        ("trend|local|seasonal",
            partial(trend, local_model=True, seasonal=True)),

        ("trend|log", partial(trend, use_log=True)),
        ("trend|log|local", partial(trend, use_log=True, local_model=True)),
        ("trend|log|seasonal", partial(trend, use_log=True, seasonal=True)),
        ("trend|log|local|seasonal",
            partial(trend, use_log=True, local_model=True, seasonal=True)),

        ("exsmooth|alpha=0.2", partial(exsmooth, alpha=0.2)),
        ("exsmooth|seasonal|alpha=0.2",
            partial(exsmooth, alpha=0.2, seasonal=True)),
        ("exsmooth|local|alpha=0.2",
            partial(exsmooth, alpha=0.2, local_model=True)),
        ("exsmooth|local|seasonal|alpha=0.2",
            partial(exsmooth, alpha=0.2, local_model=True, seasonal=True)),

        ("exsmooth|log|alpha=0.2", partial(exsmooth, alpha=0.2, use_log=True)),
        ("exsmooth|log|seasonal|alpha=0.2",
            partial(exsmooth, alpha=0.2, seasonal=True, use_log=True)),
        ("exsmooth|log|local|alpha=0.2",
            partial(exsmooth, alpha=0.2, local_model=True, use_log=True)),
        ("exsmooth|log|local|seasonal|alpha=0.2",
            partial(exsmooth, alpha=0.2, local_model=True, seasonal=True, use_log=True)),

        ("exsmooth|alpha=0.4", partial(exsmooth, alpha=0.4)),
        ("exsmooth|seasonal|alpha=0.4",
            partial(exsmooth, alpha=0.4, seasonal=True)),
        ("exsmooth|local|alpha=0.4",
            partial(exsmooth, alpha=0.4, local_model=True)),
        ("exsmooth|local|seasonal|alpha=0.4",
            partial(exsmooth, alpha=0.4, local_model=True, seasonal=True)),

        ("exsmooth|log|alpha=0.4", partial(exsmooth, alpha=0.4, use_log=True)),
        ("exsmooth|log|seasonal|alpha=0.4",
            partial(exsmooth, alpha=0.4, seasonal=True, use_log=True)),
        ("exsmooth|log|local|alpha=0.4",
            partial(exsmooth, alpha=0.4, local_model=True, use_log=True)),
        ("exsmooth|log|local|seasonal|alpha=0.4",
            partial(exsmooth, alpha=0.4, local_model=True, seasonal=True, use_log=True)),

        ("exsmooth|alpha=0.6", partial(exsmooth, alpha=0.6)),
        ("exsmooth|seasonal|alpha=0.6",
            partial(exsmooth, alpha=0.6, seasonal=True)),
        ("exsmooth|local|alpha=0.6",
            partial(exsmooth, alpha=0.6, local_model=True)),
        ("exsmooth|local|seasonal|alpha=0.6",
            partial(exsmooth, alpha=0.6, local_model=True, seasonal=True)),

        ("exsmooth|log|alpha=0.6", partial(exsmooth, alpha=0.6, use_log=True)),
        ("exsmooth|log|seasonal|alpha=0.6",
            partial(exsmooth, alpha=0.6, seasonal=True, use_log=True)),
        ("exsmooth|log|local|alpha=0.6",
            partial(exsmooth, alpha=0.6, local_model=True, use_log=True)),
        ("exsmooth|log|local|seasonal|alpha=0.6",
            partial(exsmooth, alpha=0.6, local_model=True, seasonal=True, use_log=True)),

        ("exsmooth|alpha=0.8", partial(exsmooth, alpha=0.8)),
        ("exsmooth|seasonal|alpha=0.8",
            partial(exsmooth, alpha=0.8, seasonal=True)),
        ("exsmooth|local|alpha=0.8",
            partial(exsmooth, alpha=0.8, local_model=True)),
        ("exsmooth|local|seasonal|alpha=0.8",
            partial(exsmooth, alpha=0.8, local_model=True, seasonal=True)),

        ("exsmooth|alpha=0.9", partial(exsmooth, alpha=0.9)),
        ("exsmooth|seasonal|alpha=0.9",
            partial(exsmooth, alpha=0.9, seasonal=True)),
        ("exsmooth|local|alpha=0.9",
            partial(exsmooth, alpha=0.9, local_model=True)),
        ("exsmooth|local|seasonal|alpha=0.9",
            partial(exsmooth, alpha=0.9, local_model=True, seasonal=True)),

        ("holt|alpha=0.2|beta=0.2", partial(holt, alpha=0.2, beta=0.2)),
        ("holt|seasonal|alpha=0.2|beta=0.2",
            partial(holt, alpha=0.2, beta=0.2, seasonal=True)),
        ("holt|local|alpha=0.2|beta=0.2",
            partial(holt, alpha=0.2, beta=0.2, local_model=True)),
        ("holt|local|seasonal|alpha=0.2|beta=0.2",
            partial(holt, alpha=0.2, beta=0.2, local_model=True, seasonal=True)),

        ("holt|alpha=0.4|beta=0.2", partial(holt, alpha=0.4, beta=0.2)),
        ("holt|seasonal|alpha=0.4|beta=0.2",
            partial(holt, alpha=0.4, beta=0.2, seasonal=True)),
        ("holt|local|alpha=0.4|beta=0.2",
            partial(holt, alpha=0.4, beta=0.2, local_model=True)),
        ("holt|local|seasonal|alpha=0.4|beta=0.2",
            partial(holt, alpha=0.4, beta=0.2, local_model=True, seasonal=True)),

        ("holt|alpha=0.6|beta=0.2", partial(holt, alpha=0.6, beta=0.2)),
        ("holt|seasonal|alpha=0.6|beta=0.2",
            partial(holt, alpha=0.6, beta=0.2, seasonal=True)),
        ("holt|local|alpha=0.6|beta=0.2",
            partial(holt, alpha=0.6, beta=0.2, local_model=True)),
        ("holt|local|seasonal|alpha=0.6|beta=0.2",
            partial(holt, alpha=0.6, beta=0.2, local_model=True, seasonal=True)),

        ("holt|alpha=0.2|beta=0.5",
            partial(holt, alpha=0.2, beta=0.5)),
        ("holt|seasonal|alpha=0.2|beta=0.5",
            partial(holt, alpha=0.2, beta=0.5, seasonal=True)),
        ("holt|local|alpha=0.2|beta=0.5",
            partial(holt, alpha=0.2, beta=0.5, local_model=True)),
        ("holt|local|seasonal|alpha=0.2|beta=0.5",
            partial(holt, alpha=0.2, beta=0.5, local_model=True, seasonal=True)),

        ("holt|alpha=0.4|beta=0.5", partial(holt, alpha=0.4, beta=0.5)),
        ("holt|seasonal|alpha=0.4|beta=0.5",
            partial(holt, alpha=0.4, beta=0.5, seasonal=True)),
        ("holt|local|alpha=0.4|beta=0.5",
            partial(holt, alpha=0.4, beta=0.5, local_model=True)),
        ("holt|local|seasonal|alpha=0.4|beta=0.5",
            partial(holt, alpha=0.4, beta=0.5, local_model=True, seasonal=True)),

        ("holt|alpha=0.6|beta=0.5", partial(holt, alpha=0.6, beta=0.5)),
        ("holt|seasonal|alpha=0.6|beta=0.5",
            partial(holt, alpha=0.6, beta=0.5, seasonal=True)),
        ("holt|local|alpha=0.6|beta=0.5",
            partial(holt, alpha=0.6, beta=0.5, local_model=True)),
        ("holt|local|seasonal|alpha=0.6|beta=0.5",
            partial(holt, alpha=0.6, beta=0.5, local_model=True, seasonal=True)),

        ("holt|alpha=0.8|beta=0.5", partial(holt, alpha=0.8, beta=0.5)),
        ("holt|seasonal|alpha=0.8|beta=0.5",
            partial(holt, alpha=0.8, beta=0.5, seasonal=True)),
        ("holt|local|alpha=0.8|beta=0.5",
            partial(holt, alpha=0.8, beta=0.5, local_model=True)),
        ("holt|local|seasonal|alpha=0.8|beta=0.5",
            partial(holt, alpha=0.8, beta=0.5, local_model=True, seasonal=True)),

        ("arima|001", partial(arima, q=0, d=0, p=1)),
        ("arima|001|local", partial(arima, q=0, d=0, p=1, local_model=True)),
        ("arima|001|seasonal", partial(arima, q=0, d=0, p=1, seasonal=True)),
        ("arima|001|seasonal|local", partial(arima, q=0, d=0, p=1,
            local_model=True, seasonal=True)),

        ("arima|101", partial(arima, q=1, d=0, p=1)),
        ("arima|101|local", partial(arima, q=1, d=0, p=1, local_model=True)),
        ("arima|101|seasonal", partial(arima, q=1, d=0, p=1, seasonal=True)),
        ("arima|101|seasonal|local", partial(arima, q=1, d=0, p=1,
            local_model=True, seasonal=True)),

        ("arima|201", partial(arima, q=2, d=0, p=1)),
        ("arima|201|local", partial(arima, q=2, d=0, p=1, local_model=True)),
        ("arima|201|seasonal", partial(arima, q=2, d=0, p=1, seasonal=True)),
        ("arima|201|seasonal|local", partial(arima, q=2, d=0, p=1,
            local_model=True, seasonal=True)),

        ("arima|log|201", partial(arima, q=2, d=0, p=1, use_log=True)),
        ("arima|log|201|local", partial(arima, q=2, d=0, p=1, local_model=True, use_log=True)),
        ("arima|log|201|seasonal", partial(arima, q=2, d=0, p=1, seasonal=True, use_log=True)),
        ("arima|log|201|seasonal|local", partial(arima, q=2, d=0, p=1,
            local_model=True, seasonal=True, use_log=True)),

        ("fourier", partial(fourier)),
        ("fourier|local", partial(fourier, local_model=True)),
        ("fourier|seasonal", partial(fourier, seasonal=True)),
        ("fourier|seasonal|local", partial(fourier, local_model=True, seasonal=True)),

        ("fourier|log", partial(fourier, use_log=True)),
        ("fourier|log|local", partial(fourier, local_model=True, use_log=True)),
        ("fourier|log|seasonal", partial(fourier, seasonal=True, use_log=True)),
        ("fourier|log|seasonal|local", partial(fourier, local_model=True, seasonal=True, use_log=True)),
    ]

    return grid