Skip to content

uct tree search + supervised lerning for atari games

Notifications You must be signed in to change notification settings

5vision/uct_atari

Repository files navigation

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages