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