R/helper.R

In LucasKookUZH/ontram-pkg: Ordinal transformation model neural networks

# Internal for checks

#' @importFrom tensorflow tf
.switch_method <- function(method) {
  ret <- switch(method, "logit" = tf$sigmoid,
                "cloglog" = tf_function(pgompertz),
                "loglog" = tf_function(pgumbel),
                "probit" = stop("Not implemented yet."))
  return(ret)
}

#' Get weights by name of layer
#' @export
get_weights_by_name <- function(m, name) {
  layer_names <- unlist(lapply(m$layers, function(x) x$name))
  layer_weights <- lapply(m$layers, function(x) x$weights)
  idx <- which(layer_names == name)
  as.vector(layer_weights[[idx]][[1]]$numpy())
}

.rm_int <- function(x) {
  if (all(x[, 1] == 1))
    return(x[, -1L, drop = FALSE])
  return(x)
}