dcgym / iroko

A platform to test reinforcement learning policies in the datacenter setting.
Apache License 2.0
67 stars 22 forks source link

New/custom RL algorithm #41

Open cexu2 opened 1 year ago

cexu2 commented 1 year ago

How do I add a custom RL algorithm? is there a file I need to modify ? a class that I need to implement? Any documentation will be great! I am new to gym and ray/rllib.

fruffy commented 1 year ago

Iroko is basically just a wrapper around OpenAI gym. To try out a new RL algorithm you would interface the same way as you would with the Gym environment. For example: https://github.com/dcgym/iroko/blob/master/run_basic.py is a basic test. https://github.com/dcgym/iroko/blob/master/run_ray.py interfaces with the Ray framework.

Keep in mind that this code is severely out of date.