From 05eacc3fa8712f0d7bc1605d6a00f69c75c268d3 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Wed, 10 Jul 2019 16:11:52 -0400
Subject: [PATCH] updated norm function for observation
---
utils/observation_utils.py | 10 +++++++---
1 file changed, 7 insertions(+), 3 deletions(-)
diff --git a/utils/observation_utils.py b/utils/observation_utils.py
index 0c97b18..fda5b53 100644
--- a/utils/observation_utils.py
+++ b/utils/observation_utils.py
@@ -29,7 +29,7 @@ def min_lt(seq, val):
return min
-def norm_obs_clip(obs, clip_min=-1, clip_max=1):
+def norm_obs_clip(obs, clip_min=-1, clip_max=1, fixed_radius=0):
"""
This function returns the difference between min and max value of an observation
:param obs: Observation that should be normalized
@@ -37,8 +37,12 @@ def norm_obs_clip(obs, clip_min=-1, clip_max=1):
:param clip_max: max value where observation will be clipped
:return: returnes normalized and clipped observatoin
"""
- max_obs = max(1, max_lt(obs, 1000))
- min_obs = min(max_obs, min_lt(obs, 0))
+ if fixed_radius > 0:
+ max_obs = fixed_radius
+ else:
+ max_obs = max(1, max_lt(obs, 1000))
+
+ min_obs = 0 # min(max_obs, min_lt(obs, 0))
if max_obs == min_obs:
return np.clip(np.array(obs) / max_obs, clip_min, clip_max)
--
GitLab