How do I split a custom dataset into training and test datasets?

by

Starting in PyTorch 0.4.1 you can use random_split:

train_size = int(0.8 * len(full_dataset))
test_size = len(full_dataset) - train_size
train_dataset, test_dataset = torch.utils.data.random_split(full_dataset, [train_size, test_size])

More Related Contents:

  1. Best way to save a trained model in PyTorch? [closed]
  2. How do I save a trained model in PyTorch?
  3. How to initialize weights in PyTorch?
  4. How do I initialize weights in PyTorch?
  5. Why do we need to call zero_grad() in PyTorch?
  6. How does Pytorch’s “Fold” and “Unfold” work?
  7. What does model.eval() do in pytorch?
  8. PyTorch NotImplementedError in forward
  9. Pytorch – RuntimeError: Trying to backward through the graph a second time, but the buffers have already been freed
  10. What’s the difference between torch.stack() and torch.cat() functions?
  11. RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
  12. RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! when resuming training
  13. PyTorch Binary Classification – same network structure, ‘simpler’ data, but worse performance?
  14. Understanding Keras LSTMs
  15. How to get the location of all text present in an image using OpenCV?
  16. Keras Sequential model input layer
  17. Heroku: slug size too large after installing Pytorch
  18. Does Any one got “AttributeError: ‘str’ object has no attribute ‘decode’ ” , while Loading a Keras Saved Model
  19. Keras replacing input layer
  20. How can I use a pre-trained neural network with grayscale images?
  21. Removing then Inserting a New Middle Layer in a Keras Model
  22. How to feed caffe multi label data in HDF5 format?
  23. Unbalanced data and weighted cross entropy
  24. What is the role of TimeDistributed layer in Keras?
  25. ModuleNotFoundError: No module named ‘tools.nnwrap’
  26. Keras LSTM input dimension setting
  27. Keras, how do I predict after I trained a model?
  28. looking for source code of from gen_nn_ops in tensorflow
  29. How to construct a network with two inputs in PyTorch
  30. How can I fix this strange error: “RuntimeError: CUDA error: out of memory”?

Leave a Comment