class_docstr: A test model for testing tabular hot-flips on regression tasks.
class_docstr_tok: ["A","test","model","for","testing","tabular","hot-flips","on","regression","tasks","."]
code: def predict_minibatch(
      self, inputs: list[lit_types.JsonDict], **unused
  ) -> list[lit_types.JsonDict]:
      output = []
  
      def predict_example(ex: lit_types.JsonDict) -> lit_types.JsonDict:
          x1 = ex["x_1"]
          x2 = ex["x_2"]
          return 2 * x1**2 + x2
  
      for example in inputs:
          output.append({"score": predict_example(example)})
      return output
  
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 4 for calling the RegressionTestModel obj's underlying member method code with example usage: obj.predict_minibatch(inputs) and returns: output, unknown