-
Notifications
You must be signed in to change notification settings - Fork 2
/
README
19 lines (12 loc) · 1.06 KB
/
README
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
1) Because of the large size of training data, we couldn't uplaod it here,
however you can download quantum_HEOM package from https://github.com/jwa7/quantum_HEOM,
and generate the training data. We provide the script LTLME.py
2) The farthes_point.py samples the trajectories based on farthest point sampling
(we just need to do it for one case, initial exciation on site-1 or site-6)
3) We choose 500 trajectories for site-1 and 500 trajectories from site-6, in total 1000 trajectories as a training set
4) The validation set is the 100 trajectories from site-1 + 100 trajectories from site-6
5) use prep_input.py to prepare your input files accordingly
6) use train_CNN.py to train the CNN model
7) Run run_dyn.py to predict EET dynamics for test trajectories. We have already provided our trained model "trained_ML_model.hdf5".
The respective parameters of test trajectories are in temperature.npy, gamma.npy, initial_site.npy and lambda.npy.
8) The search_optim_eet.py predict population of site-3 for different combinations of gamma, lambda and temperature.