Enhance RMSE calculation function

AlexeyPechnikov · Jun 13, 2024 · 944ff95 · 944ff95
1 parent e86a7b7
commit 944ff95
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pygmtsar/pygmtsar/Stack_sbas.py b/pygmtsar/pygmtsar/Stack_sbas.py
@@ -943,7 +943,9 @@ def rmse(self, data, solution, weight=None):
             error_pairs.append(error_pair**2)
         # form 3D stack
         error = xr.concat(error_pairs, dim='pair').assign_coords({'pair': pairs.pair})
-        return np.sqrt((weight * error).sum('pair') / weight.sum('pair') / len(pairs))
+        if weight is not None:
+            return np.sqrt((weight * error).sum('pair') / weight.sum('pair') / len(pairs)).rename('rmse')
+        return np.sqrt((error).sum('pair') / len(pairs)).rename('rmse')
 
     def plot_displacement(self, data, caption='Cumulative LOS Displacement, [rad]',
                           quantile=None, vmin=None, vmax=None, symmetrical=False, aspect=None, **kwargs):