├── .gitignore ├── README.md ├── core.py ├── dqn.py ├── env.py ├── figs ├── boxing.png └── breakout.png ├── logger.py ├── main.py ├── model.py ├── policy.py ├── roms ├── air_raid.bin ├── alien.bin ├── amidar.bin ├── assault.bin ├── asterix.bin ├── asteroids.bin ├── atlantis.bin ├── bank_heist.bin ├── battle_zone.bin ├── beam_rider.bin ├── berzerk.bin ├── bowling.bin ├── boxing.bin ├── breakout.bin ├── carnival.bin ├── centipede.bin ├── chopper_command.bin ├── crazy_climber.bin ├── defender.bin ├── demon_attack.bin ├── double_dunk.bin ├── elevator_action.bin ├── enduro.bin ├── fishing_derby.bin ├── freeway.bin ├── frostbite.bin ├── gopher.bin ├── gravitar.bin ├── hero.bin ├── ice_hockey.bin ├── jamesbond.bin ├── journey_escape.bin ├── kangaroo.bin ├── krull.bin ├── kung_fu_master.bin ├── montezuma_revenge.bin ├── ms_pacman.bin ├── name_this_game.bin ├── phoenix.bin ├── pitfall.bin ├── pong.bin ├── pooyan.bin ├── private_eye.bin ├── qbert.bin ├── riverraid.bin ├── road_runner.bin ├── robotank.bin ├── seaquest.bin ├── skiing.bin ├── solaris.bin ├── space_invaders.bin ├── star_gunner.bin ├── tennis.bin ├── time_pilot.bin ├── tutankham.bin ├── up_n_down.bin ├── venture.bin ├── video_pinball.bin ├── wizard_of_wor.bin ├── yars_revenge.bin └── zaxxon.bin ├── train.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/README.md -------------------------------------------------------------------------------- /core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/core.py -------------------------------------------------------------------------------- /dqn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/dqn.py -------------------------------------------------------------------------------- /env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/env.py -------------------------------------------------------------------------------- /figs/boxing.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/figs/boxing.png -------------------------------------------------------------------------------- /figs/breakout.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/figs/breakout.png -------------------------------------------------------------------------------- /logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/logger.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/main.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/model.py -------------------------------------------------------------------------------- /policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/policy.py -------------------------------------------------------------------------------- /roms/air_raid.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/air_raid.bin -------------------------------------------------------------------------------- /roms/alien.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/alien.bin -------------------------------------------------------------------------------- /roms/amidar.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/amidar.bin -------------------------------------------------------------------------------- /roms/assault.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/assault.bin -------------------------------------------------------------------------------- /roms/asterix.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/asterix.bin -------------------------------------------------------------------------------- /roms/asteroids.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/asteroids.bin -------------------------------------------------------------------------------- /roms/atlantis.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/atlantis.bin -------------------------------------------------------------------------------- /roms/bank_heist.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/bank_heist.bin -------------------------------------------------------------------------------- /roms/battle_zone.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/battle_zone.bin -------------------------------------------------------------------------------- /roms/beam_rider.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/beam_rider.bin -------------------------------------------------------------------------------- /roms/berzerk.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/berzerk.bin -------------------------------------------------------------------------------- /roms/bowling.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/bowling.bin -------------------------------------------------------------------------------- /roms/boxing.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/boxing.bin -------------------------------------------------------------------------------- /roms/breakout.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/breakout.bin -------------------------------------------------------------------------------- /roms/carnival.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/carnival.bin -------------------------------------------------------------------------------- /roms/centipede.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/centipede.bin -------------------------------------------------------------------------------- /roms/chopper_command.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/chopper_command.bin -------------------------------------------------------------------------------- /roms/crazy_climber.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/crazy_climber.bin -------------------------------------------------------------------------------- /roms/defender.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/defender.bin -------------------------------------------------------------------------------- /roms/demon_attack.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/demon_attack.bin -------------------------------------------------------------------------------- /roms/double_dunk.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/double_dunk.bin -------------------------------------------------------------------------------- /roms/elevator_action.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/elevator_action.bin -------------------------------------------------------------------------------- /roms/enduro.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/enduro.bin -------------------------------------------------------------------------------- /roms/fishing_derby.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/fishing_derby.bin -------------------------------------------------------------------------------- /roms/freeway.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/freeway.bin -------------------------------------------------------------------------------- /roms/frostbite.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/frostbite.bin -------------------------------------------------------------------------------- /roms/gopher.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/gopher.bin -------------------------------------------------------------------------------- /roms/gravitar.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/gravitar.bin -------------------------------------------------------------------------------- /roms/hero.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/hero.bin -------------------------------------------------------------------------------- /roms/ice_hockey.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/ice_hockey.bin -------------------------------------------------------------------------------- /roms/jamesbond.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/jamesbond.bin -------------------------------------------------------------------------------- /roms/journey_escape.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/journey_escape.bin -------------------------------------------------------------------------------- /roms/kangaroo.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/kangaroo.bin -------------------------------------------------------------------------------- /roms/krull.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/krull.bin -------------------------------------------------------------------------------- /roms/kung_fu_master.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/kung_fu_master.bin -------------------------------------------------------------------------------- /roms/montezuma_revenge.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/montezuma_revenge.bin -------------------------------------------------------------------------------- /roms/ms_pacman.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/ms_pacman.bin -------------------------------------------------------------------------------- /roms/name_this_game.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/name_this_game.bin -------------------------------------------------------------------------------- /roms/phoenix.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/phoenix.bin -------------------------------------------------------------------------------- /roms/pitfall.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/pitfall.bin -------------------------------------------------------------------------------- /roms/pong.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/pong.bin -------------------------------------------------------------------------------- /roms/pooyan.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/pooyan.bin -------------------------------------------------------------------------------- /roms/private_eye.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/private_eye.bin -------------------------------------------------------------------------------- /roms/qbert.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/qbert.bin -------------------------------------------------------------------------------- /roms/riverraid.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/riverraid.bin -------------------------------------------------------------------------------- /roms/road_runner.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/road_runner.bin -------------------------------------------------------------------------------- /roms/robotank.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/robotank.bin -------------------------------------------------------------------------------- /roms/seaquest.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/seaquest.bin -------------------------------------------------------------------------------- /roms/skiing.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/skiing.bin -------------------------------------------------------------------------------- /roms/solaris.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/solaris.bin -------------------------------------------------------------------------------- /roms/space_invaders.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/space_invaders.bin -------------------------------------------------------------------------------- /roms/star_gunner.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/star_gunner.bin -------------------------------------------------------------------------------- /roms/tennis.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/tennis.bin -------------------------------------------------------------------------------- /roms/time_pilot.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/time_pilot.bin -------------------------------------------------------------------------------- /roms/tutankham.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/tutankham.bin -------------------------------------------------------------------------------- /roms/up_n_down.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/up_n_down.bin -------------------------------------------------------------------------------- /roms/venture.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/venture.bin -------------------------------------------------------------------------------- /roms/video_pinball.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/video_pinball.bin -------------------------------------------------------------------------------- /roms/wizard_of_wor.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/wizard_of_wor.bin -------------------------------------------------------------------------------- /roms/yars_revenge.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/yars_revenge.bin -------------------------------------------------------------------------------- /roms/zaxxon.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/roms/zaxxon.bin -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hengyuan-hu/rainbow/HEAD/utils.py --------------------------------------------------------------------------------