From 512c1dc03bb0dba4aea2f258d445ff07e728f249 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 20 May 2022 14:20:17 -0500
Subject: [PATCH] add method to scale data

---
 modules/util/util.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/modules/util/util.py b/modules/util/util.py
index 24983e1f..0ccb0c14 100644
--- a/modules/util/util.py
+++ b/modules/util/util.py
@@ -405,6 +405,29 @@ def normalize(data, param, mean_std_dict, add_noise=False, noise_scale=1.0, seed
     return data
 
 
+def scale(data, param, mean_std_dict):
+
+    if mean_std_dict.get(param) is None:
+        return data
+
+    shape = data.shape
+    data = data.flatten()
+
+    mean, std, lo, hi = mean_std_dict.get(param)
+    data -= mean
+    data /= std
+
+    data -= lo
+    data /= (hi - lo)
+
+    not_valid = np.isnan(data)
+    data[not_valid] = 0
+
+    data = np.reshape(data, shape)
+
+    return data
+
+
 f = open(ancillary_path+'geos_crs_goes16_FD.pkl', 'rb')
 geos_goes16_fd = pickle.load(f)
 f.close()
-- 
GitLab