train_agent.py contains the training code for a specific agent. The plot_train_agent.ipynb contains the visulization of a pre-trained agent.
This project are more foucsed on the analysis of a given trained model, and do model analysis based on the provided model.