AminHP / gym-anytrading

The most simple, flexible, and comprehensive OpenAI Gym trading environment (Approved by OpenAI Gym)
MIT License
2.1k stars 465 forks source link

stable-basellines3 example #33

Closed highbulb closed 3 years ago

highbulb commented 3 years ago

It will be very helpful if you can point me some examples using stable-baselines3. I am still not sure how it comparable to stable-baselines as they have big warning box to compare performance. Appreciated.

AminHP commented 3 years ago

According to this table, recurrent networks are not supported yet for A2C. You should use other policies like MlpPolicy. To make this example work with stable-baselines3, put the code below in the section Train Env:

model = A2C('MlpPolicy', env, verbose=1)
model.learn(total_timesteps=1000)