'fix-bug-r-square'

superarthurlx · Jan 2, 2025 · e554d9f · e554d9f
1 parent e716d1d
commit e554d9f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/basicts/metrics/r_square.py b/basicts/metrics/r_square.py
@@ -39,3 +39,4 @@ def masked_r2(prediction: torch.Tensor, target: torch.Tensor, null_val: float =
 
     # 计算 R^2
     loss = 1 - (ss_res / (ss_tot + eps))
+     return torch.mean(loss)