deepof.utils.filter_short_bouts

deepof.utils.filter_short_bouts(cluster_assignments: ndarray, cluster_confidence: ndarray, confidence_indices: ndarray, min_confidence: float = 0.0, min_bout_duration: int | None = None)

Filter out cluster assignment bouts shorter than min_bout_duration.

Parameters:

• cluster_assignments (np.ndarray) – Array of cluster assignments.

• cluster_confidence (np.ndarray) – Array of cluster confidence values.

• confidence_indices (np.ndarray) – Array of confidence indices.

• min_confidence (float) – Minimum confidence value.

• min_bout_duration (int) – Minimum bout duration in frames.

Returns:

Mask of confidence indices to keep.

Return type:

np.ndarray