The Main Codes folder contains all codes for train the model. -forward_train.py trains a maximum likelihood forward model. -RI_train performs a immature reinforcement training. The current prediction of the model is used as the baseline. -A2C.py employs an advantage actor-critic algorithm.
The myGUI folder contains files for the GUI. -classifer.py is the main program