PyTorch

PyTorchでデータセットを学習、検証、テストと分割したい場合は、torch.utils.data.random_splitを用います。引数datasetに分割するデータセット、第二引数lengthsに分割後のデータ数、第三引数generatorはオプションでデータの分割に再現性を持たせたい場合は指定する。