register env's

2026-08-12 06:20:11 +08:00 · 2017-10-29 13:29:47 +08:00
parent daaf23fe83
commit d0a9f9900d
1 changed files with 31 additions and 0 deletions
@@ -0,0 +1,31 @@
+import pandas as pd
+from gym.envs.registration import register
+
+df_train = pd.read_hdf('./data/poloniex_30m.hf', key='train')
+
+register(
+    id='PortfolioMLP-v0',
+    entry_point='src.environments.portfolio:PortfolioEnv',
+    kwargs=dict(
+        output_mode='mlp',
+        df=df_train
+    )
+)
+
+register(
+    id='PortfolioEIIE-v0',
+    entry_point='src.environments.portfolio:PortfolioEnv',
+    kwargs=dict(
+        output_mode='EIIE',
+        df=df_train
+    )
+)
+
+register(
+    id='PortfolioAtari-v0',
+    entry_point='src.environments.portfolio:PortfolioEnv',
+    kwargs=dict(
+        output_mode='atari',
+        df=df_train
+    )
+)