5vision / uct_atari

uct tree search + supervised lerning for atari games

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

UCT search for atari games

This repo contains code for running uct search on atari games, collecting uct search data in form (frame, best_uct_action) and then use it in supervised training.

UCT search

To run just uct search use:

$ python run_uct.py

To run uct search and collect data for further training use:

$ python collect_uct_data.py

Supervised learning

To run supervised training with collected data use:

$ python train_keras.py dirs_with_uct_data

Use flag --help to see all options for UCT and supervised learning scripts.

About

uct tree search + supervised lerning for atari games


Languages

Language:Python 100.0%