R/outcome-family.R

In ck37/tlmixture: Data-adaptive Creation Of Exposure (Treatment) Mixtures Using Targeted Learning

#' Detect outcome family based on the outcome data vector
#'
#' @param data tbd
#' @param outcome_field tbd
#'
#' @export
outcome_family = function(data, outcome_field) {

  unique_vals = unique(data[[outcome_field]])
  if (length(unique_vals) == 2L &&
      length(setdiff(unique(data[[outcome_field]]), c(0, 1))) == 0L) {
    family = "binomial"
  } else {
    family = "gaussian"
  }

  return(family)

}