chainer.datasets.split_dataset

chainer.datasets.split_dataset(dataset, split_at, order=None)[source]

Splits a dataset into two subsets.

This function creates two instances of SubDataset. These instances do not share any examples, and they together cover all examples of the original dataset.

Parameters
  • dataset – Dataset to split.

  • split_at (int) – Position at which the base dataset is split.

  • order (sequence of ints) – Permutation of indexes in the base dataset. See the documentation of SubDataset for details.

Returns

Two SubDataset objects. The first subset represents the examples of indexes order[:split_at] while the second subset represents the examples of indexes order[split_at:].

Return type

tuple