code: def __init__(
      self,
      train_set: Dataset,
      train_batch_size: int,
      val_set: Dataset,
      val_batch_size: int,
      test_set: Dataset = None,
      test_batch_size: int = None,
  ):
      self.train_set: Dataset = train_set
      self.train_batch_size: int = train_batch_size
  
      self.val_set: Dataset = val_set
      self.val_batch_size: int = val_batch_size
  
      self.test_set: Dataset = test_set
      self.test_batch_size: int = test_batch_size
  
file_path: transfer-nlp/transfer_nlp/loaders/loaders.py
imports: from torch.utils.data import Dataset, DataLoader
label_desc: Use image node_id 1 to create a new DatasetSplits object  with example: obj = DatasetSplits(train_set, train_batch_size, val_set, val_batch_size, test_set, test_batch_size)