code: def __init__(
      self, batch_size, test_batch_size, train_dataset, model, sampler
  ):
      self._batch_size = batch_size
      self._test_batch_size = test_batch_size
  
      self._train_dataset = train_dataset
      self._max_item = self._train_dataset.max_item()
  
      self._model = model
      self._sampler = sampler
  
file_path: openrec/openrec/tf1/legacy/itr_mlp_model_trainer.py
imports: from __future__ import print_function
  from tqdm import tqdm
  import math
  from termcolor import colored
  import numpy
  import sys
label_desc: Use image node_id 1 to create a new ItrMLPModelTrainer object from inherited base classes: object with example: obj = ItrMLPModelTrainer(batch_size, test_batch_size, train_dataset, model, sampler)