change model

XAMS-nikhef · Apr 24, 2024 · 7ee2a88 · 7ee2a88
1 parent 12460bd
commit 7ee2a88
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/amstrax/plugins/peaks/peaks_som.py b/amstrax/plugins/peaks/peaks_som.py
@@ -35,7 +35,7 @@ class PeaksSOM(Peaks):
 
     # This is not a great solution but needed for now
     directory = os.path.dirname(__file__)  # Get the directory where the script is located
-    file_path = os.path.join(directory, 'som_models', 'xams_som_v1.npz')
+    file_path = os.path.join(directory, 'som_models', 'xams_som_v2.npz')
     som_files = np.load(file_path)
 
     use_som_as_default = True

diff --git a/amstrax/plugins/peaks/som_models/xams_som_v2.npz b/amstrax/plugins/peaks/som_models/xams_som_v2.npz