Separate 2 groups in Cox regression

Instalation

if (!require("BiocManager")) {
    install.packages("BiocManager")
}
BiocManager::install("glmSparseNet")

Required Packages

library(futile.logger)
library(ggplot2)
library(glmSparseNet)
library(survival)

# Some general options for futile.logger the debugging package
flog.layout(layout.format("[~l] ~m"))
options("glmSparseNet.show_message" = FALSE)
# Setting ggplot2 default theme as minimal
theme_set(ggplot2::theme_minimal())

Prepare data

data("cancer", package = "survival")
xdata <- survival::ovarian[, c("age", "resid.ds")]
ydata <- data.frame(
    time = survival::ovarian$futime,
    status = survival::ovarian$fustat
)

Separate using age as co-variate

(group cutoff is median calculated relative risk)

resAge <- separate2GroupsCox(c(age = 1, 0), xdata, ydata)

Kaplan-Meier survival results

## Call: survfit(formula = survival::Surv(time, status) ~ group, data = prognosticIndexDf)
## 
##                n events median 0.95LCL 0.95UCL
## Low risk - 1  13      4     NA     638      NA
## High risk - 1 13      8    464     268      NA

Plot

A individual is attributed to low-risk group if its calculated relative risk (using Cox Proportional model) is below or equal the median risk.

The opposite for the high-risk groups, populated with individuals above the median relative-risk.

Separate using age as co-variate (group cutoff is 40% - 60%)

resAge4060 <-
    separate2GroupsCox(c(age = 1, 0),
        xdata,
        ydata,
        probs = c(.4, .6)
    )

Kaplan-Meier survival results

## Call: survfit(formula = survival::Surv(time, status) ~ group, data = prognosticIndexDf)
## 
##                n events median 0.95LCL 0.95UCL
## Low risk - 1  11      3     NA     563      NA
## High risk - 1 10      7    359     156      NA

Plot

A individual is attributed to low-risk group if its calculated relative risk (using Cox Proportional model) is below the median risk.

The opposite for the high-risk groups, populated with individuals above the median relative-risk.

Separate using age as co-variate (group cutoff is 60% - 40%)

This is a special case where you want to use a cutoff that includes some sample on both high and low risks groups.

resAge6040 <- separate2GroupsCox(
    chosenBetas = c(age = 1, 0),
    xdata,
    ydata,
    probs = c(.6, .4),
    stopWhenOverlap = FALSE
)
## Warning in buildPrognosticIndexDataFrame(ydata, probs, stopWhenOverlap, : The cutoff values given to the function allow for some over samples in both groups, with:
##   high risk size (15) + low risk size (16) not equal to xdata/ydata rows (31 != 26)
## 
## We are continuing with execution as parameter `stopWhenOverlap` is FALSE.
##   note: This adds duplicate samples to ydata and xdata xdata

Kaplan-Meier survival results

## Kaplan-Meier results
## Call: survfit(formula = survival::Surv(time, status) ~ group, data = prognosticIndexDf)
## 
##                n events median 0.95LCL 0.95UCL
## Low risk - 1  16      5     NA     638      NA
## High risk - 1 15      9    475     353      NA

Plot

A individual is attributed to low-risk group if its calculated relative risk (using Cox Proportional model) is below the median risk.

The opposite for the high-risk groups, populated with individuals above the median relative-risk.

Session Info

sessionInfo()
## R version 4.6.2 (2026-10-29)
## Platform: x86_64-pc-linux-gnu
## Running under: Ubuntu 26.04.1 LTS
## 
## Matrix products: default
## BLAS:   /usr/lib/x86_64-linux-gnu/openblas-pthread/libblas.so.3 
## LAPACK: /usr/lib/x86_64-linux-gnu/openblas-pthread/libopenblasp-r0.3.32.so;  LAPACK version 3.12.0
## 
## locale:
##  [1] LC_CTYPE=en_US.UTF-8       LC_NUMERIC=C              
##  [3] LC_TIME=en_US.UTF-8        LC_COLLATE=en_US.UTF-8    
##  [5] LC_MONETARY=en_US.UTF-8    LC_MESSAGES=en_US.UTF-8   
##  [7] LC_PAPER=en_US.UTF-8       LC_NAME=C                 
##  [9] LC_ADDRESS=C               LC_TELEPHONE=C            
## [11] LC_MEASUREMENT=en_US.UTF-8 LC_IDENTIFICATION=C       
## 
## time zone: Etc/UTC
## tzcode source: system (glibc)
## 
## attached base packages:
##  [1] grid      parallel  stats4    stats     graphics  grDevices utils    
##  [8] datasets  methods   base     
## 
## other attached packages:
##  [1] VennDiagram_1.8.2           reshape2_1.4.5             
##  [3] forcats_1.0.1               Matrix_1.7-6               
##  [5] glmSparseNet_1.31.0         TCGAutils_1.33.3           
##  [7] curatedTCGAData_1.35.1      MultiAssayExperiment_1.39.1
##  [9] SummarizedExperiment_1.43.0 Biobase_2.73.2             
## [11] GenomicRanges_1.65.4        Seqinfo_1.3.2              
## [13] IRanges_2.47.5              S4Vectors_0.51.10          
## [15] BiocGenerics_0.59.12        generics_0.1.4             
## [17] MatrixGenerics_1.25.0       matrixStats_1.5.0          
## [19] futile.logger_1.4.9         survival_3.8-12            
## [21] ggplot2_4.0.3               dplyr_1.2.1                
## [23] BiocStyle_2.41.0           
## 
## loaded via a namespace (and not attached):
##   [1] RColorBrewer_1.1-3        sys_3.4.3                
##   [3] jsonlite_2.0.0            magrittr_2.0.5           
##   [5] GenomicFeatures_1.65.0    farver_2.1.2             
##   [7] rmarkdown_2.32            BiocIO_1.23.3            
##   [9] vctrs_0.7.3               memoise_2.0.1            
##  [11] Rsamtools_2.29.0          RCurl_1.98-1.20          
##  [13] rstatix_1.1.0             htmltools_0.5.9          
##  [15] S4Arrays_1.13.2           BiocBaseUtils_1.15.1     
##  [17] AnnotationHub_4.3.3       lambda.r_1.2.6           
##  [19] curl_8.1.0                broom_1.0.13             
##  [21] Formula_1.2-6             SparseArray_1.13.4       
##  [23] sass_0.4.10               bslib_0.12.0             
##  [25] plyr_1.8.9                httr2_1.3.0              
##  [27] futile.options_1.0.1      cachem_1.1.0             
##  [29] buildtools_1.0.0          GenomicAlignments_1.49.2 
##  [31] lifecycle_1.0.5           pkgconfig_2.0.3          
##  [33] R6_2.6.1                  fastmap_1.2.0            
##  [35] digest_0.6.39             AnnotationDbi_1.75.2     
##  [37] ExperimentHub_3.3.2       RSQLite_3.53.3           
##  [39] ggpubr_1.0.0              filelock_1.0.3           
##  [41] labeling_0.4.3            httr_1.4.9               
##  [43] abind_1.4-8               compiler_4.6.2           
##  [45] bit64_4.8.6               withr_3.0.3              
##  [47] S7_0.2.2                  backports_1.5.1          
##  [49] BiocParallel_1.47.0       carData_3.0-6            
##  [51] DBI_1.3.0                 ggsignif_0.6.4           
##  [53] rappdirs_0.3.4            DelayedArray_0.39.8      
##  [55] rjson_0.2.23              tools_4.6.2              
##  [57] otel_0.2.0                glue_1.8.1               
##  [59] restfulr_0.0.17           checkmate_2.3.4          
##  [61] gtable_0.3.6              tzdb_0.5.0               
##  [63] tidyr_1.3.2               survminer_0.5.2          
##  [65] hms_1.1.4                 xml2_1.6.0               
##  [67] car_3.1-5                 XVector_0.53.0           
##  [69] BiocVersion_3.24.0        pillar_1.11.1            
##  [71] stringr_1.6.0             splines_4.6.2            
##  [73] BiocFileCache_3.3.1       lattice_0.23-1           
##  [75] rtracklayer_1.73.0        bit_4.6.0                
##  [77] tidyselect_1.2.1          maketools_1.3.2          
##  [79] Biostrings_2.81.9         knitr_1.52               
##  [81] gridExtra_2.3.1           xfun_0.62                
##  [83] stringi_1.8.9             UCSC.utils_1.9.0         
##  [85] yaml_2.3.12               evaluate_1.0.5           
##  [87] codetools_0.2-20          cigarillo_1.3.1          
##  [89] tibble_3.3.1              BiocManager_1.30.27      
##  [91] cli_3.6.6                 jquerylib_0.1.4          
##  [93] Rcpp_1.1.2                GenomeInfoDb_1.49.1      
##  [95] GenomicDataCommons_1.37.0 dbplyr_2.6.0             
##  [97] png_0.1-9                 XML_3.99-0.25            
##  [99] readr_2.2.0               blob_1.3.0               
## [101] bitops_1.1-0              scales_1.4.0             
## [103] purrr_1.2.2               crayon_1.5.3             
## [105] rlang_1.3.0               KEGGREST_1.53.6          
## [107] rvest_1.0.5               formatR_1.14