Fit an ARIMAX model to one orgUnit — fit_arima_OneOrgUnit • PRIDEC

Fit an ARIMAX model to one orgUnit

Usage

fit_arima_OneOrgUnit(
  train_df,
  test_df,
  pred_vars,
  quantile_levels,
  return_model = FALSE,
  log_trans = FALSE
)

Arguments

train_df: dataframe of analysis data
test_df: dataframe of assessment data
pred_vars: predictor variables to include
quantile_levels: quantiles needed for prediction intervals (assessment only)
return_model: whether or not to return the ARIMA model, mostly used for debugging. Default = FALSE
log_trans: whether to log transform the data to help with fit. Uses log(y+1). Default = FALSE

Value

dataframe of prediction intervals for analysis and assessment data