Extract data from an ml object — get_data • martini

Combine split data (training and test, if available) from either prepared or raw data

Usage

get_data(ml_obj, type = c("prep", "raw"), split_id = NULL)

Arguments

ml_obj: ml object as returned by prepare_ml()
type: either prep or raw, determining which state of the data should be extracted. Defaults to prep.
split_id: column name (character). Add column indicating split origin (train/test). Omitted if NULL (default).

Value

result of dplyr::bind_rows() of data sets in ml_obj of the chosen type, either with or without an added train_test column.

Authors

Maike Ahrens (ahrensmaike), Sebastian Voss (svoss09)