make_test_train: Make Test and Train identifiers

Description Usage Arguments

Description

Assigns rows to test/train designations depending on fold_col/fold_id combination. If null, test and train are the entire dataset.

Usage

1
make_test_train(data, fold_col = NULL, fold_id = NULL)

Arguments

data

data.table. Dataset to be split into test and train. Requires a column called 'rid' which is the row id.

fold_col

character vector. Name of a column in data from which the test/train split should be made

fold_id

numeric vector. Value designating which rows (given fold_col) should be left out.


dahcase/mbgstacking documentation built on May 20, 2019, 4:08 p.m.