Skip to content

Commit 134ccaf

Browse files
afermgalxndrkalinin
authored andcommitted
fix(multilabel): replace find_pairs->find_pairs_multilabel
1 parent 6fc68e2 commit 134ccaf

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/copairs/map/multilabel.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
import pandas as pd
88

99
from copairs import compute
10-
from copairs.matching import UnpairedException, find_pairs
10+
from copairs.matching import UnpairedException, find_pairs_multilabel
1111

1212
from .filter import evaluate_and_filter, flatten_str_list, validate_pipeline_input
1313

@@ -95,13 +95,13 @@ def average_precision(
9595
logger.info("Indexing metadata...")
9696

9797
logger.info("Finding positive pairs...")
98-
pos_pairs = find_pairs(meta, sameby=pos_sameby, diffby=pos_diffby)
98+
pos_pairs = find_pairs_multilabel(meta, sameby=pos_sameby, diffby=pos_diffby, multilabel_col=multilabel_col)
9999
if len(pos_pairs) == 0:
100100
raise UnpairedException("Unable to find positive pairs.")
101101

102102
logger.info("Finding negative pairs...")
103103
_, pos_counts = np.unique(pos_pairs, axis=0, return_counts=True)
104-
neg_pairs = find_pairs(meta, sameby=neg_sameby, diffby=neg_diffby)
104+
neg_pairs = find_pairs_multilabel(meta, sameby=neg_sameby, diffby=neg_diffby, multilabel_col=multilabel_col)
105105
if len(neg_pairs) == 0:
106106
raise UnpairedException("Unable to find any negative pairs.")
107107

0 commit comments

Comments
 (0)