R/supportedMethods.R

In ericdunipace/causalOT: Optimal Transport Weights for Causal Inference

#' Supported Methods
#'
#' @return A character list with supported methods. Note "COT" is the same as "Wasserstein". We provide the second name for backwards compatibility.
#' @export
#'
#' @examples
#' supported_methods()
supported_methods <- function() {
  c(sort(c("Wasserstein", grid_search_methods())), likelihood_methods())
}

likelihood_methods <- function() {
  c("Logistic","Probit","CBPS")
}

grid_search_methods <- function() {
  c(balancing_function_methods(), balancing_distributions_methods())
}

balancing_function_methods <- function() {
  c("SBW","EntropyBW")
}

balancing_distributions_methods <- function() {
  c("SCM", cot_methods())
}

cot_methods <- function() {
  c("COT","EnergyBW", "NNM")
}