From 0567e53b75d641eabc27a413e31392107202367d Mon Sep 17 00:00:00 2001
From: Danilo Ferreira de Lima <danilo.enoque.ferreira.de.lima@xfel.de>
Date: Sun, 8 Oct 2023 11:28:00 +0200
Subject: [PATCH] Reset PCA comps.
---
pes_to_spec/test/offline_analysis.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/pes_to_spec/test/offline_analysis.py b/pes_to_spec/test/offline_analysis.py
index 11966b7..ed80b67 100755
--- a/pes_to_spec/test/offline_analysis.py
+++ b/pes_to_spec/test/offline_analysis.py
@@ -289,14 +289,14 @@ def main():
pca = PCA(None, whiten=True)
pca.fit(pes_raw_select)
df = pd.DataFrame(dict(variance_ratio=pca.explained_variance_ratio_,
- n_comp=1000*np.ones_like(pca.explained_variance_ratio_),
+ n_comp=600*np.ones_like(pca.explained_variance_ratio_),
))
df.to_csv(os.path.join(args.directory, "pca_pes.csv"))
pca_spec = PCA(None, whiten=True)
pca_spec.fit(spec_raw_int[train_idx])
df = pd.DataFrame(dict(variance_ratio=pca_spec.explained_variance_ratio_,
- n_comp=40*np.ones_like(pca_spec.explained_variance_ratio_),
+ n_comp=20*np.ones_like(pca_spec.explained_variance_ratio_),
))
df.to_csv(os.path.join(args.directory, "pca_spec.csv"))
--
GitLab