Merge pull request #70 from cinziabraglia7/main

Column names bug fix in output table

Author: leomccarthy

R/output_table.R

@@ -29,10 +29,9 @@ output_table <- function(data, model) {
     segmentation_vars <- allcolumnnames[!allcolumnnames %in% c('id', response , 'segment')]  
   }
 
-  df_agg <- df %>% select(c('segment',model$model_hyperparameters$segmentation_variables)) 
-  characterlevel <- lapply(df_agg,is.character)==TRUE
-  
-  df_agg_numeric <- df_agg[, unlist(lapply(df_agg, is.numeric)) | names(df_agg) == 'segment', drop = FALSE] %>%
+  df_agg <- df %>% select(c('segment',model$model_hyperparameters$segmentation_variables))
+
+  df_agg_numeric <- df_agg[, unlist(lapply(df_agg, is.numeric)) | names(df_agg) == 'segment'] %>%
     group_by(.data$segment) %>%
     summarise(across(everything(), ~round(mean(.data$., na.rm = TRUE), 2)))
 
@@ -41,6 +40,7 @@ output_table <- function(data, model) {
     summarise(across(everything(), ~mode(.data$.)))
 
   df_agg <- full_join(df_agg_numeric, df_agg_character, by = 'segment')
+  characterlevel <- lapply(df_agg,is.character)==TRUE
 
   names(df_agg)[characterlevel] <- paste0(names(df_agg)[characterlevel],'_mode')
   names(df_agg)[!characterlevel] <- paste0(names(df_agg)[!characterlevel],'_mean')