estimate_qra_qenspy: Estimate qra model using qenspy package as backend

In reichlab/covidEnsembles: What the Package Does (One Line, Title Case)

Estimate qra model using qenspy package as backend

Description

Estimate qra model using qenspy package as backend

Usage

estimate_qra_qenspy(
  qfm_train,
  y_train,
  combine_method,
  quantile_groups,
  init_param_vec = NULL,
  optim_method = "adam",
  num_iter = 1000,
  learning_rate = 0.1,
  verbose = FALSE,
  partial_save_frequency,
  partial_save_filename
)

Arguments

qfm_train	QuantileForecastMatrix with training set predictions
y_train	numeric vector of responses for training set
quantile_groups	Vector of group labels for quantiles, having the same length as the number of quantiles. Common labels indicate that the ensemble weights for the corresponding quantile levels should be tied together. Default is rep(1,length(quantiles)), which means that a common set of ensemble weights should be used across all levels. This is the argument tau_groups for quantgen::quantile_ensemble
qfm_test	QuantileForecastMatrix with test set predictions
intercept	logical specifying whether an intercept is included
constraint	character specifying constraints on parameters; 'convex', 'positive' or 'unconstrained'

Value

object of class qra_fit

reichlab/covidEnsembles documentation built on Jan. 31, 2024, 7:21 p.m.