Corrected expected value of Gamma-Bernoulli distribution.

climax/core/predict.py

@@ -67,7 +67,7 @@ def predict_ERA5(net, ERA5_ds, predictand, batch_size=16, **kwargs):
         prob = torch.sigmoid(torch.as_tensor(target[:, 0, ...].squeeze(),
                                              dtype=torch.float32)).numpy()
 
-        # amount of precipitation: expected value of Bernoulli-Gamma distribution
+        # precipitation amount: expected value of Bernoulli-Gamma distribution
         # pr = p * shape * scale
         pr = (prob * np.exp(target[:, 1, ...].squeeze()) *
               np.exp(target[:, 2, ...].squeeze()))