code: def val_dataloader(self):
      return DataLoader(self.val_set, batch_size=4)
  
file_path: mlflow/examples/pytorch/AxHyperOptimizationPTL/iris_data_module.py
imports: import pytorch_lightning
  import torch
  from sklearn.datasets import load_iris
  from torch.utils.data import DataLoader, TensorDataset, random_split
label_desc: Use image node_id 5 for calling the IrisDataModule obj's underlying member method code with example usage: obj.val_dataloader() and returns: DataLoader