code: def on_epoch_end(self):
      self._counter = 0
      sss = StratifiedShuffleSplit(
          n_splits=self._n_splits,
          train_size=self._max_size,
          test_size=self._validation_size,
      )
      self._splits = sss.split(self._x, self._y)
  
file_path: AIX360/aix360/algorithms/nncontrastive/utils.py
imports: import math
  import numpy
  from tensorflow.keras.utils import Sequence
  from sklearn.model_selection import StratifiedShuffleSplit
label_desc: Use image node_id 2 for calling the JointAESampler obj's underlying member method code with example usage: obj.on_epoch_end() without return types