Add return value and reorder columns.

Author: ntthung

R/mass-balance-optim.R

@@ -118,6 +118,7 @@ back_trans <- function(hat, years, mus, sigmas, log.trans, N, season.names) {
 #' @param lambda The penalty weight
 #' @param log.trans A vector containing indices of the targets to be log-transformed. If no transformation is needed, provide `NULL`.
 #' @param force.standardize If TRUE, all observations are standardized. See Details.
+#' @return A `data.table` with the following columns: season, year, Q, and lambda.
 #' @section Details:
 #' If some targets are log transformed and some are not, they will have different scales, which affects the objective function. In this case the observations will be standardized so that they are in the same range. Otherwise, standardization are skipped for speed. However, in some cases you may want to standardize any ways, for example when flows in some months are much larger than in other months. In this case, set `force.standardize = TRUE`.
 #' @examples
@@ -198,6 +199,8 @@ mb_reconstruction <- function(instQ, pc.list, start.year, lambda = 1,
   hat <- X %*% beta
   DT <- back_trans(hat, years, cm, csd, log.trans, N, seasons)
   DT[, lambda := lambda][]
+  setcolorder(DT, c('season', 'year', 'Q', 'lambda'))
+  DT[]
 }