Esempi in Python per retrieval_average_precision

Esempio n. 1

File: mean_average_precision.py Progetto: chenchy/metrics

 def _metric(self, preds: Tensor, target: Tensor) -> Tensor:
     valid_indexes = (target != self.exclude)
     return retrieval_average_precision(preds[valid_indexes],
                                        target[valid_indexes])

Esempio n. 2

 def _metric(self, preds: Tensor, target: Tensor) -> Tensor:
     return retrieval_average_precision(preds, target)