h2o (version 3.10.5.3)

h2o.kfold_column: Produce a k-fold column vector.

Description

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

Usage

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.