lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-01-11 09:29:33 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Drop duplicate bin edges

Browse Source
This commit is contained in:
mbsantiago 2025-08-11 08:59:12 +01:00
parent f361420e58
commit 31a0d1bbb5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/batdetect2/plotting/evaluation.py
View File

@ -154,7 +154,7 @@ def get_binned_sample(matches: List[MatchEvaluation], n_examples: int = 5):
] ]
) )
bins = pd.qcut(pred_scores, q=n_examples, labels=False) bins = pd.qcut(pred_scores, q=n_examples, labels=False, duplicates="drop")
df = pd.DataFrame({"indices": indices, "bins": bins}) df = pd.DataFrame({"indices": indices, "bins": bins})
sample = df.groupby("bins").apply(lambda x: x.sample(1)) sample = df.groupby("bins").apply(lambda x: x.sample(1))
return [matches[ind] for ind in sample["indices"]] return [matches[ind] for ind in sample["indices"]]