R/helper_functions.R

In dustinfife/flexplot: Graphically Based Data Analysis Using 'flexplot'

return_factors_names = function(model) {
  
  terms = remove_interaction_terms(model)
  d = extract_data_from_fitted_object(model)
  d %>% select(-where(is.numeric)) %>%  names
}

identify_method = function(data, outcome, axis, method=NULL) {
  # histograms/barcharts
  if (axis[1] == "1") return("loess")
  if (!is.null(method)) return(method) 
  # association plot
  if (check.non.number(data[,axis[1]])) return("loess")
  # logistic
  if (length(unique(data[,outcome]))==2) return("logistic")
   
  return("loess")
}