bacow/easy-rl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix Q-leaning

Browse Source
This commit is contained in:
JohnJim0816
2021-02-27 16:58:40 +08:00
parent fd84a59875
commit 23a9cf2cba
1 changed files with 0 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
codes/Q-learning/main.py
View File

@@ -136,11 +136,6 @@ def eval(cfg, saved_model_path = SAVED_MODEL_PATH):
plot(rewards) plot(rewards)
plot(MA_rewards,ylabel='moving_average_rewards_train') plot(MA_rewards,ylabel='moving_average_rewards_train')
def main():
cfg = get_args()
# train(cfg)
test(cfg)
if __name__ == "__main__": if __name__ == "__main__":
cfg = get_args() cfg = get_args()
if cfg.train: if cfg.train: