datasets

prosit_data_merge.py

@@ -150,7 +150,8 @@ df = pd.read_pickle('database/data_prosit_merged_holdout.pkl')
 
 print(len(df))
 print(df.head())
-df = df[df['Retention time']!=[0,0,0,0,0,0]]
+df['Retention time']=df['Retention time'].apply(lambda x : x[0])
+df = df.loc[df['Retention time']!=0]
 print(len(df))
 print(df.head())