chainer_chemistry.dataset.splitters.StratifiedSplitter

class chainer_chemistry.dataset.splitters.StratifiedSplitter

Class for doing stratified data splits.

__init__()

Initialize self. See help(type(self)) for accurate signature.

Methods

k_fold_split(dataset, k)
train_valid_split(dataset[, labels, …])	Split dataset into train and valid set.
train_valid_test_split(dataset[, labels, …])	Split dataset into train, valid and test set.