StratifiedPartition: Augment cohort table with a stratified train/test partition

In mbadge/hipsMultimodal: Analysis of MSH Hip Data

Description

this fxn wraps createDataPartition

Usage

StratifiedPartition(cohort_df, grp_col = "pt", target_col = "fx")

Arguments

cohort_df	data.frame
grp_col	chr(1)
target_col	chr(1)

Value

data.frame with same number of rows and an additional column 'partition' with values train and test

Examples

StratifiedPartition(mtcars, grp_col="am", target_col="mpg") %>% s
StratifiedPartition(mtcars, grp_col="am", target_col="mpg") %>% select(am, partition) %>% table_()
StratifiedPartition(hipsCohort(), grp_col = "device_model") %>% select(device_model, partition) %>% table_()
StratifiedPartition(hipsCohort()) %>% select(pt, partition) %>% table_()
try(StratifiedPartition(hipsCohort() %>% sample_n(500)) %>% select(pt, partition) %>% table_())
StratifiedPartition(hipsCohort()) %>% attributes() %>% s

mbadge/hipsMultimodal documentation built on May 9, 2019, 12:05 a.m.