From b6d130730a7e2ae641d415cac739b04561fa47f5 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 22 Oct 2024 14:18:42 +0200
Subject: [PATCH] datasets

---
 prosit_data_merge.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/prosit_data_merge.py b/prosit_data_merge.py
index 7a612c7..02b48fd 100644
--- a/prosit_data_merge.py
+++ b/prosit_data_merge.py
@@ -151,10 +151,12 @@ print(len(df))
 possible_charges = [1,2,3,4]
 df = df[df['Charge'].isin(possible_charges)]
 print(len(df))
+df.to_pickle('database/data_prosit_merged_holdout.pkl')
 df = pd.read_pickle('database/data_prosit_merged_train.pkl')
 print(len(df))
 df=df.loc[df['Retention time']!=0]
 df = df[df['Charge'].isin(possible_charges)]
 print(len(df))
+df.to_pickle('database/data_prosit_merged_train.pkl')
 
 
-- 
GitLab