prep_data: ——————- Prepare dataset

In treeheatr: Heatmap-Integrated Decision Tree Visualizations

Description

———————————————————————————— Prepare dataset

Usage

prep_data(data, target_lab, task, feat_types = NULL)

Arguments

data	Original data frame with features to be converted to correct types.
target_lab	Name of the column in data that contains target/label information.
task	Character string indicating the type of problem, either 'classification' (categorical outcome) or 'regression' (continuous outcome).
feat_types	Named vector indicating the type of each features, e.g., c(sex = 'factor', age = 'numeric'). If feature types are not supplied, infer from column type.

Value

List of dataframes (training + test) with proper feature types and target name.

treeheatr documentation built on Nov. 20, 2020, 1:07 a.m.