From 99dd774be15994eb421093d36c0730e262316991 Mon Sep 17 00:00:00 2001 From: PaoloBonettiPolimi <94172434+PaoloBonettiPolimi@users.noreply.github.com> Date: Thu, 25 Apr 2024 10:56:14 +0200 Subject: [PATCH] Update main.py --- hawk/analysis/main.py | 17 ----------------- 1 file changed, 17 deletions(-) diff --git a/hawk/analysis/main.py b/hawk/analysis/main.py index ae888d5..1f5bad9 100644 --- a/hawk/analysis/main.py +++ b/hawk/analysis/main.py @@ -83,16 +83,12 @@ def run_baseline_analysis(self): features_names = self.datasets["normal"]["var_names"] - print(f"features_names: {features_names}") - configs = [] # Only autoregressive baselines from 1 to the maximum target lag for i in range(1, self.tefs_max_lag_target+1): configs.append((f"AR({i})", {self.target_column_name: list(range(1, i + 1))})) - print(f"AR configs: {configs}") - # All features without AR configs.append( ( @@ -101,8 +97,6 @@ def run_baseline_analysis(self): ) ) - print(f"Full configs: {configs}") - # All features with AR configs.append( ( @@ -112,8 +106,6 @@ def run_baseline_analysis(self): ) ) - print(f"Full configs: {configs}") - for label, inputs_names_lags in configs: baseline[label] = { "inputs": inputs_names_lags, @@ -125,8 +117,6 @@ def run_baseline_analysis(self): ), } - print(f"Baselines: {baseline}") - target_file = os.path.join(self.workdir, "baseline.pkl") save_to_pkl_file(target_file, baseline) @@ -172,8 +162,6 @@ def run_tefs_analysis( } configurations.append(configuration) - print(f"TEFS configurations: {configurations}") - # Run the analysis results = [] for config in configurations: @@ -185,8 +173,6 @@ def run_tefs_analysis( ) ) - print(f"TEFS results: {results}") - return results def run_pcmci_analysis( @@ -240,8 +226,6 @@ def run_pcmci_analysis( } configurations.append(configuration) - print(f'PCMCI configurations: {configurations}') - # Run the analysis results = [] for config in configurations: @@ -252,7 +236,6 @@ def run_pcmci_analysis( independence_tests=independence_tests, ) ) - print(f"PCMCI+ results: {results}") return results