From d2e0dac8c85a9a7f8cf2c2519358d3187b3e12d6 Mon Sep 17 00:00:00 2001 From: Erikpostt Date: Mon, 23 Sep 2024 08:50:23 +0200 Subject: [PATCH] naming convention update --- src/paradigma/windowing.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/src/paradigma/windowing.py b/src/paradigma/windowing.py index 555ee06..f3cca53 100644 --- a/src/paradigma/windowing.py +++ b/src/paradigma/windowing.py @@ -135,20 +135,20 @@ def discard_segments(df, segment_nr_colname, min_length_segment_s, sampling_freq return filtered_df -def categorize_segments(df, segment_nr_colname, config): +def categorize_segments(df, segment_nr_colname, sampling_frequency): # Calculate the number of rows corresponding to 5, 10, and 20 seconds - short_segments_max = 5 * config.sampling_frequency # 5 seconds - moderately_long_segments_max = 10 * config.sampling_frequency # 10 seconds - long_segments_max = 20 * config.sampling_frequency # 20 seconds + short_segments_max_duration = 5 * sampling_frequency # 5 seconds + moderately_long_segments_max_duration = 10 * sampling_frequency # 10 seconds + long_segments_max_duration = 20 * sampling_frequency # 20 seconds # Group by the segment column and apply the categorization def categorize(group): segment_size = len(group) - if segment_size < short_segments_max: + if segment_size < short_segments_max_duration: return 1 - elif segment_size < moderately_long_segments_max: + elif segment_size < moderately_long_segments_max_duration: return 2 - elif segment_size < long_segments_max: + elif segment_size < long_segments_max_duration: return 3 else: return 4