src.evaluation_metric.matching_precision

src.evaluation_metric.matching_precision(pmat_pred: torch.Tensor, pmat_gt: torch.Tensor, ns: torch.Tensor) torch.Tensor[source]

Matching Precision between predicted permutation matrix and ground truth permutation matrix.

\[\text{matching precision} = \frac{tr(\mathbf{X}\cdot {\mathbf{X}^{gt}}^\top)}{\sum \mathbf{X}}\]
Parameters
  • pmat_pred\((b\times n_1 \times n_2)\) predicted permutation matrix \((\mathbf{X})\)

  • pmat_gt\((b\times n_1 \times n_2)\) ground truth permutation matrix \((\mathbf{X}^{gt})\)

  • ns\((b)\) number of exact pairs. We support batched instances with different number of nodes, and ns is required to specify the exact number of nodes of each instance in the batch.

Returns

\((b)\) matching precision

Note

This function is equivalent to “matching accuracy” if the matching problem has no outliers.