From 6e8b8f0913cc6014c2f58fc3277ff673006eff7c Mon Sep 17 00:00:00 2001
From: bin_open_source <342748465@qq.com>
Date: Tue, 3 Oct 2023 23:19:32 +0800
Subject: [PATCH] ADD file via upload

---
 utils/metrics.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 utils/metrics.py

diff --git a/utils/metrics.py b/utils/metrics.py
new file mode 100644
index 0000000..8f974f2
--- /dev/null
+++ b/utils/metrics.py
@@ -0,0 +1,33 @@
+import numpy as np
+
+def RSE(pred, true):
+    return np.sqrt(np.sum((true-pred)**2)) / np.sqrt(np.sum((true-true.mean())**2))
+
+def CORR(pred, true):
+    u = ((true-true.mean(0))*(pred-pred.mean(0))).sum(0) 
+    d = np.sqrt(((true-true.mean(0))**2*(pred-pred.mean(0))**2).sum(0))
+    return (u/d).mean(-1)
+
+def MAE(pred, true):
+    return np.mean(np.abs(pred-true))
+
+def MSE(pred, true):
+    return np.mean((pred-true)**2)
+
+def RMSE(pred, true):
+    return np.sqrt(MSE(pred, true))
+
+def MAPE(pred, true):
+    return np.mean(np.abs((pred - true) / true))
+
+def MSPE(pred, true):
+    return np.mean(np.square((pred - true) / true))
+
+def metric(pred, true):
+    mae = MAE(pred, true)
+    mse = MSE(pred, true)
+    rmse = RMSE(pred, true)
+    mape = MAPE(pred, true)
+    mspe = MSPE(pred, true)
+    
+    return mae,mse,rmse,mape,mspe
\ No newline at end of file