class_docstr: A class to load dataset for ranking task.
class_docstr_tok: ["A","class","to","load","dataset","for","ranking","task","."]
docstr: Returns a tf.dataset.Dataset.
docstr_tok: ["Returns","a","tf.dataset.Dataset","."]
code: def load(
      self, input_context: Optional[tf.distribute.InputContext] = None
  ) -> tf.data.Dataset:
      """Returns a tf.dataset.Dataset."""
      reader = input_reader.InputReader(
          params=self._params,
          dataset_fn=self._dataset_fn,
          decoder_fn=self._decode,
          parser_fn=self._parse,
      )
      return reader.read(input_context)
  
imports: import dataclasses
  from typing import Callable, Dict, Mapping, Optional, Tuple, Union
  import tensorflow
  from tensorflow_ranking.python import data
  from tensorflow_ranking.python.keras import losses
  from tensorflow_ranking.python.keras import metrics
  from tensorflow_ranking.python.keras import model
label_desc: Use image node_id 4 for calling the RankingDataLoader obj's underlying member method code with example usage: obj.load(input_context) and returns: reader