Create a k-fold vector useful for H2O algorithms that take a fold_assignments argument.

h2o.kfold_column(data, nfolds, seed = -1)

Arguments

data

A dataframe against which to create the fold column.

nfolds

The number of desired folds.

seed

A random seed, -1 indicates that H2O will choose one.

Value

Returns an H2OFrame object with fold assignments.