shaneucsd / wt4elegantrl

Wt4ElegantRL

Wt4ElegantRL = WonderTrader + ElegantRL

Document by panyunan

https://gitee.com/panyunan/wt4elegantrl-doc

INSTALL

clone

git clone https://github.com/drlgistics/Wt4ElegantRL.git
cd Wt4ElegantRL

with cuda

conda env create -n wt4elegantrl --file ./requirements/full_with_cuda.yaml

or without cuda

conda env create -n wt4elegantrl --file ./requirements/full_without_cuda.yaml

activate

conda activate wt4elegantrl

RLLIB

demo

python ./compare_rllib.py test -p ./trained/rllib/TD3_2021-10-29_11-21-37/TD3_SimpleCTAEnv_536e7_00000_0_2021-10-29_11-21-37/checkpoint_000023/checkpoint-23

train

python ./compare_rllib.py train

SB3

demo

python ./compare_sb3.py test -p ./trained/sb3/td3/best_model

train

python ./compare_sb3.py train

ELEGANTRL

train

python ./compare_elegantrl.py train