class_docstr: Tests tabular hot-flips on classification tasks.
class_docstr_tok: ["Tests","tabular","hot-flips","on","classification","tasks","."]
docstr: Given an example, returns the index of the top prediction.
code: def _predict_and_return_argmax_label(self, example):
      """Given an example, returns the index of the top prediction."""
      model_out = self._model.predict([example])
      softmax = list(model_out)[0]["preds"]
      argmax = np.argmax(softmax)
      return self._model.output_spec()["preds"].vocab[argmax]
  
imports: from unittest import mock
  from absl.testing import absltest
  from lit_nlp.api import dataset
  from lit_nlp.api import model
  from lit_nlp.api import types
  from lit_nlp.components import minimal_targeted_counterfactuals
  from lit_nlp.lib import caching
  import numpy
  import scipy.special
label_desc: Use image node_id 13 for calling the ClassificationTabularMtcTest obj's underlying member method code with example usage: obj._predict_and_return_argmax_label(example) and returns: self