lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-04-04 23:30:21 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Run formatting

Browse Source
This commit is contained in:
mbsantiago 2026-03-08 08:59:28 +00:00
parent 8313fe1484
commit cce1b49a8d
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/batdetect2/data/split.py
View File

@ -28,10 +28,12 @@ def split_dataset_by_recordings(
majority_class = ( majority_class = (
sound_events.groupby("recording_id") sound_events.groupby("recording_id")
.apply( .apply(
lambda group: group["class_name"] # type: ignore lambda group: (
.value_counts() group["class_name"] # type: ignore
.sort_values(ascending=False) .value_counts()
.index[0], .sort_values(ascending=False)
.index[0]
),
include_groups=False, # type: ignore include_groups=False, # type: ignore
) )
.rename("class_name") .rename("class_name")