busy week update

jurjen93 · Feb 13, 2024 · d14d5e9 · d14d5e9
1 parent 3322b23
commit d14d5e9
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git 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
@@ -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