diff --git a/data_processing.py b/data_processing.py index 41102ff..2bdd942 100644 --- a/data_processing.py +++ b/data_processing.py @@ -35,6 +35,7 @@ def compute_correlation_matrix(data): trimmed_data = [d[:min_length] for d in data] corr_matrix = np.corrcoef(trimmed_data) + np.fill_diagonal(corr_matrix, 0) return corr_matrix def split_data_by_time(data, split_ratio=0.5):