Bug fix in eps calculation.

pes_to_spec/model.py

@@ -114,7 +114,7 @@ class Model(object):
         low_pca = self.lr_pca.fit_transform(low_res)
         high_pca = self.hr_pca.fit_transform(high_res)
         # split in train and test for PCA uncertainty evaluation
-        low_pca_train, low_pca_test, high_pca_train, high_pca_test = train_test_split(low_pca, high_pca, test_size=self.test_size)
+        low_pca_train, low_pca_test, high_pca_train, high_pca_test = train_test_split(low_pca, high_pca, test_size=self.test_size, random_state=42)
         # fit the linear model
         self.fit_model.fit(low_pca_train, high_pca_train, low_pca_test, high_pca_test)
 

scripts/test_analysis.py

@@ -87,7 +87,7 @@ def main():
     # plot
     for tid in test_tids:
         idx = np.where(tid==tids)[0][0]
-        plot_result(f"test_{tid}.png", spec_pred[idx, :, :], spec_smooth[idx, :], spec_raw_pe[idx, :], eps)
+        plot_result(f"test_{tid}.png", spec_pred[idx, :, :], spec_smooth[idx, :], spec_raw_pe[idx, :])
 
 if __name__ == '__main__':
     main()