diff --git a/common_dataset.py b/common_dataset.py index bab5944e439f64aedf8194928f505b2c2a0a4c10..42c4c191ae891b3217c7dd009025aa6aa5b9ab74 100644 --- a/common_dataset.py +++ b/common_dataset.py @@ -170,12 +170,12 @@ seq_train = np.load('data/intensity/sequence_train.npy') charge_train = np.load('data/intensity/precursor_charge_train.npy') spectra_train = np.load('data/intensity/intensity_train.npy') -# irt_holdout = np.load('data/intensity/collision_irt_holdout.npy') -# seq_holdout = np.load('data/intensity/sequence_holdout.npy') -# charge_holdout = np.load('data/intensity/precursor_charge_holdout.npy') -# spectra_holdout = np.load('data/intensity/intensity_holdout.npy') +irt_holdout = np.load('data/intensity/collision_irt_holdout.npy') +seq_holdout = np.load('data/intensity/sequence_holdout.npy') +charge_holdout = np.load('data/intensity/precursor_charge_holdout.npy') +spectra_holdout = np.load('data/intensity/intensity_holdout.npy') dataset_train = pd.DataFrame({'Sequence':list(seq_train), 'Retention time':list(irt_train), 'Charge':list(charge_train), 'Spectra' : list(spectra_train)},index=list(range(6787933))) dataset_train.to_pickle('database/data_prosit_merged_train.pkl') -# dataset_test = pd.DataFrame({'Sequence':list(seq_holdout), 'Retention time':list(irt_holdout), 'Charge':list(charge_holdout), 'Spectra' : list(spectra_holdout)},index=list(range(754215))) -# dataset_train.to_pickle('database/data_prosit_merged_holdout.pkl') \ No newline at end of file +dataset_test = pd.DataFrame({'Sequence':list(seq_holdout), 'Retention time':list(irt_holdout), 'Charge':list(charge_holdout), 'Spectra' : list(spectra_holdout)},index=list(range(754215))) +dataset_train.to_pickle('database/data_prosit_merged_holdout.pkl') \ No newline at end of file