class_docstr: Save mnist data into TFRecordDataset and get samples from generator
class_docstr_tok: ["Save","mnist","data","into","TFRecordDataset","and","get","samples","from","generator"]
code: def reset_batch_size(self, batch_size):
      self.batch_size = batch_size
      self.iterations = self.num_samples // self.batch_size
  
imports: import os
  import tensorflow
  from tensorflow.keras.datasets import mnist
  from .convert import decode_data
  from .convert import decode_image
  from .convert import decode_label
  from .convert import save_data
  from .convert import save_image
  from .convert import save_label
label_desc: Use image node_id 3 for calling the MnistDataset obj's underlying member method code with example usage: obj.reset_batch_size(batch_size) without return types