iai (version 1.10.2)

split_data: 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

if (FALSE) {
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
}