class_docstr: Coref model for a single edge. Compatible with EdgeFeaturesDataset.
class_docstr_tok: ["Coref","model","for","a","single","edge",".","Compatible","with","EdgeFeaturesDataset","."]
code: def output_spec(self):
      return {
          "proba": lit_types.RegressionScore(
              parent="label"
          )  # in range [0,1]
      }
  
imports: import os
  from collections.abc import Iterable
  from typing import Optional
  from absl import logging
  from lit_nlp.api import dataset
  from lit_nlp.api import dtypes
  from lit_nlp.api import model
  from lit_nlp.api import types
  from lit_nlp.lib import utils
  import numpy
  import tensorflow
label_desc: Use image node_id 10 for calling the SingleEdgePredictor obj's underlying member method code with example usage: obj.output_spec() and returns: dict