R/losses.R

Defines functions get_losses mean_square_error

get_losses <- function(loss){
  out <- switch(loss,
                "MSE" = mean_square_error,
                stop("loss function not implemented"))
}

mean_square_error <- function(y, y_pred){
  return(rowMeans((y_pred - y)^2))
}
cdeterman/prometheus documentation built on May 28, 2019, 7:16 p.m.