Ejemplo n.º 1
0
 def compute(self) -> Tensor:
     """
     Computes fbeta over state.
     """
     tp, fp, tn, fn = self._get_final_stats()
     return _fbeta_compute(tp, fp, tn, fn, self.beta, self.ignore_index,
                           self.average, self.mdmc_reduce)
Ejemplo n.º 2
0
 def compute(self) -> Tensor:
     """
     Computes fbeta over state.
     """
     return _fbeta_compute(
         self.true_positives, self.predicted_positives, self.actual_positives, self.beta, self.average
     )
Ejemplo n.º 3
0
 def compute(self):
     return _fbeta_compute(
         self.true_positives,
         self.predicted_positives,
         self.actual_positives,
         beta=1.0,
         average="weighted",
     )