Tianshou/README.md

71 lines
3.1 KiB
Markdown
Raw Normal View History

2017-11-04 01:38:59 +08:00
# tianshou
Tianshou(天授) is a reinforcement learning platform. The following image illustrate its architecture.
<img src="https://github.com/sproblvem/tianshou/blob/master/docs/figures/tianshou_architecture.png" height="200"/>
## agent
&nbsp;&nbsp;&nbsp;&nbsp;Examples
2017-12-04 16:21:33 +08:00
&nbsp;&nbsp;&nbsp;&nbsp;Self-play Framework
2017-11-06 15:17:55 +08:00
## core
### Policy Wrapper
&nbsp;&nbsp;&nbsp;&nbsp;Stochastic policies (OnehotCategorical, Gaussian), deterministic policies (policy as in DQN, DDPG)
&nbsp;&nbsp;&nbsp;&nbsp;Specific network architectures in original paper of DQN, TRPO, A3C, etc. Policy-Value Network of AlphaGo Zero
2017-11-06 15:17:55 +08:00
### Algorithm
2017-11-06 15:17:55 +08:00
#### losses
&nbsp;&nbsp;&nbsp;&nbsp;policy gradient (and its variants), DQN (and its variants), DDPG, TRPO, PPO
#### optimizer
&nbsp;&nbsp;&nbsp;&nbsp;TRPO, natural gradient (and TensorFlow optimizers (sgd, adam))
2017-11-06 20:39:09 +08:00
### Planning
&nbsp;&nbsp;&nbsp;&nbsp;MCTS
2017-11-06 20:39:09 +08:00
## data
&nbsp;&nbsp;&nbsp;&nbsp;Training style - Batch, Replay (and its variants)
&nbsp;&nbsp;&nbsp;&nbsp;Advantage Estimation Function
2017-11-06 20:39:09 +08:00
&nbsp;&nbsp;&nbsp;&nbsp;Multithread Read/Write
## environment
&nbsp;&nbsp;&nbsp;&nbsp;DQN repeat frames, Reward Reshaping, image preprocessing (not sure where)
## simulator
&nbsp;&nbsp;&nbsp;&nbsp;Go, Othello/Reversi, Warzone
<img src="https://github.com/sproblvem/tianshou/blob/master/docs/figures/go.png" height="150"/> <img src="https://github.com/sproblvem/tianshou/blob/master/docs/figures/reversi.jpg" height="150"/> <img src="https://github.com/sproblvem/tianshou/blob/master/docs/figures/warzone.jpg" height="150"/>
2017-12-10 14:23:40 +08:00
## About coding style
Please follow [google python coding style](https://google.github.io/styleguide/pyguide.html)
2017-12-10 17:33:10 +08:00
All files/folders should be named with lower case letters and underline (except specified names such as `AlphaGo`).
2017-12-10 17:33:10 +08:00
Try to use full names. Don't use abbrevations for class/function/variable names except common abbrevations (such as `num` for number, `dim` for dimension, `env` for environment, `op` for operation). For now we use `pi` to refer to the policy in examples/ppo_example.py.
The """xxx""" comment should be written right after class/function. Also comment the part that's not intuitive during the code. We must comment, but for now we don't need to polish them.
2017-12-10 13:34:07 +08:00
2017-12-10 13:36:43 +08:00
2017-12-10 14:23:40 +08:00
## TODO
Search based method parallel.
2017-12-10 13:31:43 +08:00
YongRen: Policy Wrapper, in order of Gaussian, DQN and DDPG
TongzhengRen: losses, in order of ppo, pg, DQN, DDPG with management of placeholders
YouQiaoben: data/Batch, implement num_timesteps, fix memory growth in num_episodes; adv_estimate.gae_lambda (need to write a value network in tf)
ShihongSong: data/Replay; then adv_estimate.dqn after YongRen's DQN
2017-12-08 21:18:29 +08:00
HaoshengZou: collaborate mainly on Policy and losses; interfaces and architecture
Note: install openai/gym first to run the Atari environment; note that interfaces between modules may not be finalized; the management of placeholders and `feed_dict` may have to be done manually for the time being;
Without preprocessing and other tricks, this example will not train to any meaningful results. Codes should past two tests: individual module test and run through this example code.