chainer_chemistry.dataset.splitters.StratifiedSplitter¶
-
class
chainer_chemistry.dataset.splitters.
StratifiedSplitter
[source]¶ 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. -