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

8
R/regression_model.R
View file

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