andreas/FreesearchR
1
0
Fork 0
mirror of https://github.com/agdamsbo/FreesearchR.git synced 2025-09-12 09:59:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

handling empty formula

Browse source
This commit is contained in:
Andreas Gammelgaard Damsbo 2024-11-15 21:57:38 +01:00
parent 269658ef17
commit f4be547ed0
No known key found for this signature in database
1 changed files with 9 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
R/regression_model.R
View file

@ -35,7 +35,7 @@ regression_model <- function(data,
args.list = NULL,
fun = NULL,
vars = NULL) {
if (!is.null(formula.str)) {
if (!is.null(formula.str) | formula.str != "") {
formula.str <- glue::glue(formula.str)
} else {
assertthat::assert_that(outcome.str %in% names(data),
@ -44,10 +44,10 @@ regression_model <- function(data,
formula.str <- glue::glue("{outcome.str}~.")
if (!is.null(vars)) {
if (outcome.str %in% vars){
if (outcome.str %in% vars) {
vars <- vars[vars %in% outcome.str]
}
data <- data |> dplyr::select(dplyr::all_of(c(vars,outcome.str)))
data <- data |> dplyr::select(dplyr::all_of(c(vars, outcome.str)))
}
}
@ -57,16 +57,15 @@ regression_model <- function(data,
# browser()
if (auto.mode) {
if (is.numeric(data[[outcome.str]])){
if (is.numeric(data[[outcome.str]])) {
fun <- "stats::lm"
} else if (is.factor(data[[outcome.str]])){
if (length(levels(data[[outcome.str]]))==2){
} else if (is.factor(data[[outcome.str]])) {
if (length(levels(data[[outcome.str]])) == 2) {
fun <- "stats::glm"
args.list = list(family = binomial(link = "logit"))
} else if (length(levels(data[[outcome.str]]))>2){
args.list <- list(family = binomial(link = "logit"))
} else if (length(levels(data[[outcome.str]])) > 2) {
fun <- "MASS::polr"
args.list = list(
args.list <- list(
Hess = TRUE,
method = "logistic"
)
@ -76,7 +75,6 @@ regression_model <- function(data,
} else {
stop("Output variable should be either numeric or factor for auto.mode")
}
}
assertthat::assert_that("character" %in% class(fun),