code: def _input_mappings(self, batch_data, train):
      if train:
          return {
              self._get_input("user_id"): batch_data["user_id_input"],
              self._get_input("item_id"): batch_data["item_id_input"],
              self._get_input("labels"): batch_data["labels"],
          }
      else:
          return {
              self._get_input("user_id", train=False): batch_data[
                  "user_id_input"
              ],
              self._get_input("item_id", train=False): batch_data[
                  "item_id_input"
              ],
          }
  
file_path: openrec/openrec/tf1/legacy/recommenders/pmf.py
imports: from openrec.tf1.legacy.recommenders import Recommender
  from openrec.tf1.legacy.modules.extractions import LatentFactor
  from openrec.tf1.legacy.modules.interactions import PointwiseMSE
label_desc: Use image node_id 2 for calling the PMF obj's underlying member method code with example usage: obj._input_mappings(batch_data, train) and returns: dict, dict