class_docstr: IMDB reviews dataset; see http://ai.stanford.edu/~amaas/data/sentiment/.
class_docstr_tok: ["IMDB","reviews","dataset",";","see","http",":","\/\/ai.stanford.edu\/~amaas\/data\/sentiment\/","."]
docstr: Dataset spec, which should match the model"s input_spec().
docstr_tok: ["Dataset","spec",",","which","should","match","the","model","''","s","input_spec","(",")","."]
code: def spec(self) -> lit_types.Spec:
      """Dataset spec, which should match the model"s input_spec()."""
      return {
          "text": lit_types.TextSegment(),
          "label": lit_types.CategoryLabel(vocab=self.LABELS),
      }
  
imports: from typing import Optional
  from absl import logging
  from lit_nlp.api import dataset
  from lit_nlp.api import types
  import pandas
  import tensorflow_datasets
label_desc: Use image node_id 3 for calling the IMDBData obj's underlying member method code with example usage: obj.spec() and returns: dict