Skip to content

Commit

Permalink
pass recording as input to hdbscan clustering
Browse files Browse the repository at this point in the history
  • Loading branch information
julienboussard authored Jan 17, 2024
1 parent db7ec7b commit f8527ee
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions src/dartsort/cluster/initial.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ def cluster_chunk(
clustering_config,
chunk_time_range_s=None,
motion_est=None,
recording=None,
):
"""Cluster spikes from a single segment
Expand Down Expand Up @@ -70,6 +71,7 @@ def cluster_chunk(
)
elif clustering_config.cluster_strategy == "hdbscan":
labels[in_chunk] = cluster_util.hdbscan_clustering(
recording,
times_s[in_chunk],
xyza[in_chunk, 0],
xyza[in_chunk, 2],
Expand Down Expand Up @@ -141,6 +143,7 @@ def cluster_chunks(
clustering_config,
chunk_time_range_s=chunk_range,
motion_est=motion_est,
recording=recording,
)
for chunk_range in chunk_time_ranges_s
]
Expand Down

0 comments on commit f8527ee

Please sign in to comment.