R/zzz.R

In modeltime: The Tidymodels Extension for Time Series Modeling

# IMPORTS ----
# StanHeaders - Used to prevent issues with Prophet dynload error

#' @import StanHeaders
#' @import tidymodels
NULL

# ON LOAD ----

# The functions below define the model information. These access the model
# environment inside of parsnip so they have to be executed once parsnip has
# been loaded.

.onLoad <- function(libname, pkgname) {

    # CRAN OMP THREAD LIMIT
    Sys.setenv("OMP_THREAD_LIMIT" = 1)

    # This defines the model database

    # Prophet
    make_prophet_reg()
    make_prophet_boost()

    # ARIMA
    make_arima_reg()
    make_arima_boost()

    # Exponential Smoothing
    make_exp_smoothing()

    # TBATS & SEASONAL DECOMP
    make_seasonal_reg()

    # NNETAR
    make_nnetar_reg()

    # BASELINE MODELS
    make_window_reg()
    make_naive_reg()

    #HIERARCHICAL MODEL
    make_temporal_hierarchy()

    #ADAM
    make_adam_reg()

}