bigframes.ml.model_selection#

Functions for test/train split and model tuning. This module is styled after scikit-learn’s model_selection module: https://scikit-learn.org/stable/modules/classes.html#module-sklearn.model_selection.

Functions

cast(typ, val)

Cast a value to a type.

cross_validate(estimator, X[, y, cv])

Evaluate metric(s) by cross-validation and also record fit/score times.

train_test_split(*arrays[, test_size, ...])

Splits dataframes or series into random train and test subsets.

Classes

KFold([n_splits, random_state])

K-Fold cross-validator.

chain(*iterables)

Return a chain object whose .__next__() method returns elements from the first iterable until it is exhausted, then elements from the next iterable, until all of the iterables are exhausted.