code: def __init__(self, data_feature, data_attribute):
      self.data_feature = data_feature
      self.data_attribute = data_attribute
  
file_path: BigDL/python/chronos/src/bigdl/chronos/simulator/doppelganger/data_module.py
imports: import os
  import pickle
  import numpy
  import sys
  import torch
  from torch.utils.data import DataLoader, Dataset
  from pytorch_lightning import LightningDataModule
  from .util import add_gen_flag, normalize_per_sample, renormalize_per_sample
label_desc: Use image node_id 1 to create a new CustomizedDataset object from inherited base classes: Dataset with example: obj = CustomizedDataset(data_feature, data_attribute)