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.