wassname/rl-portfolio-management
1
0
Fork 0
mirror of https://github.com/wassname/rl-portfolio-management.git synced 2026-06-27 16:46:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

concat other way round

Browse Source
This commit is contained in:
wassname
2017-11-13 07:48:25 +08:00
parent 3597f185cc
commit ea47fc42f7
2 changed files with 279 additions and 241 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rl_portfolio_management/wrappers/concat_states.py
+10 -3
View File
@@ -7,9 +7,16 @@ def concat_states(state):
history = state["history"]
weights = state["weights"]
weight_insert_shape = (history.shape[0], 1, history.shape[2])
weight_insert = np.ones(
weight_insert_shape) * weights[1:, np.newaxis, np.newaxis]
state = np.concatenate([history, weight_insert], axis=1)
if len(weights) - 1 == history.shape[0]:
weight_insert = np.ones(
weight_insert_shape) * weights[1:, np.newaxis, np.newaxis]
elif len(weights) - 1 == history.shape[2]:
weight_insert = np.ones(
weight_insert_shape) * weights[np.newaxis, np.newaxis, 1:]
else:
weight_insert = np.ones(
weight_insert_shape) * weights[np.newaxis, 1:, np.newaxis]
state = np.concatenate([weight_insert, history], axis=1)
return state
tensorforce-PPO-IEET.ipynb
+269 -238
View File
File diff suppressed because one or more lines are too long