From d14d5e972c8d7b01cd87886597724679018543e9 Mon Sep 17 00:00:00 2001 From: jurjen93 Date: Tue, 13 Feb 2024 18:57:35 +0100 Subject: [PATCH] busy week update --- source_selection/phasediff_output.py | 1 - source_selection/pre_processing_for_ml.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/source_selection/phasediff_output.py b/source_selection/phasediff_output.py index 5460ba57..d62f0fa1 100644 --- a/source_selection/phasediff_output.py +++ b/source_selection/phasediff_output.py @@ -147,7 +147,6 @@ def get_phasediff_score(self, station: str = None): H = tables.open_file(self.h5) stations = [make_utf8(s) for s in list(H.root.sol000.antenna[:]['name'])] - print(stations) if station is None or station == '': stations_idx = [stations.index(stion) for stion in stations if diff --git a/source_selection/pre_processing_for_ml.py b/source_selection/pre_processing_for_ml.py index 5027496d..dfbb8f58 100644 --- a/source_selection/pre_processing_for_ml.py +++ b/source_selection/pre_processing_for_ml.py @@ -87,7 +87,7 @@ def __init__(self, root_dir): for file in os.listdir(cls_path): if file.endswith('.fits'): self.data.append(os.path.join(cls_path, file)) - label = 'good' in cls + label = 'stop' in cls self.labels.append(int(label)) @staticmethod