Class Score

language/diffqg/metrics.py:38–62

Evaluation of a single example pair.

@dataclasses.dataclass
class Score:
  """Evaluation of a single example pair."""

  # pylint: disable=invalid-name because L is more readable here.
  rouge_L: float
  rouge_1: float
  label: Label
  bleurt: Optional[float] = None
  qsim: Optional[bool] = None

  def to_dict(self):
    """Translates class into dictionary, excluding unspecified values."""
    scores = {
        "rougeL": self.rouge_L,
        "rouge_1": self.rouge_1,
        "label": f"{self.label}",
    }
    if self.qsim is not None:
      scores["qsim"]: self.qsim
    if self.bleurt is not None:
      scores["bleurt"] = self.bleurt
    return scores

  def __str__(self):
    return json.dumps(self.to_dict())


@dataclasses.dataclass