split_data {iai}

Split the data into training and test datasets

Description

Julia Equivalent: IAI.split_data

Usage

split_data(task, X, ...)

Arguments

task

The type of problem.

X

The features of the data.

...

Other parameters, including zero or more target vectors as required by the problem type. Refer to the Julia documentation for available parameters.

Examples

## Not run: 
X <- iris[, 1:4]
y <- iris$Species
split <- iai::split_data("classification", X, y, train_proportion = 0.75)
train_X <- split$train$X
train_y <- split$train$y
test_X <- split$test$X
test_y <- split$test$y

## End(Not run)