Skip to contents

Wrapper around train_test_split_stratified for simple usage

Usage

train_test_split(X, y, test_size = 0.5, n_per_class = NULL, seed = NULL)

Arguments

X

Feature matrix (cells x features)

y

Class labels (factor or character vector)

test_size

Fraction of data for testing (default: 0.5)

n_per_class

Maximum number of samples per class in training set. If NULL, uses test_size fraction

seed

Random seed for reproducibility