From d8302405881739cf61b261f075e830e33e88702d Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Fri, 23 Jul 2021 17:39:29 +0200
Subject: [PATCH] Calculate precipitation amount from gamma distribution.
---
climax/core/predict.py | 7 +++----
1 file changed, 3 insertions(+), 4 deletions(-)
diff --git a/climax/core/predict.py b/climax/core/predict.py
index 1b6ca17..c6a93ab 100644
--- a/climax/core/predict.py
+++ b/climax/core/predict.py
@@ -63,13 +63,12 @@ def predict_ERA5(net, ERA5_ds, predictand, batch_size=16, **kwargs):
'tasmin': EoDataset.add_coordinates(target[:, 1, ...].squeeze())}
elif predictand == 'pr':
ds = {'prob': EoDataset.add_coordinates(target[:, 0, ...].squeeze()),
- 'shape': EoDataset.add_coordinates(target[:, 1, ...].squeeze()),
- 'scale': EoDataset.add_coordinates(target[:, 2, ...].squeeze()),
# amount of precipitation: expected value of gamma distribution
# pr = shape * scale
- 'pr': EoDataset.add_coordinates((target[:, 1, ...] *
- target[:, 2, ...]).squeeze())}
+ 'pr': EoDataset.add_coordinates(
+ (np.exp(target[:, 1, ...]) *
+ np.exp(target[:, 2, ...])).squeeze())}
else:
# single predictand
ds = {predictand: EoDataset.add_coordinates(target)}
--
GitLab