From 5663352cdd654bc61f8664c7e8b6994ea2790a37 Mon Sep 17 00:00:00 2001
From: Yandi <yandirzm@gmail.com>
Date: Sat, 21 Jan 2023 20:32:57 +0100
Subject: [PATCH] [Loss function] Adding the loss function

---
 losses/RMSLE.py | 9 +++++++++
 1 file changed, 9 insertions(+)
 create mode 100644 losses/RMSLE.py

diff --git a/losses/RMSLE.py b/losses/RMSLE.py
new file mode 100644
index 0000000..cf5cff6
--- /dev/null
+++ b/losses/RMSLE.py
@@ -0,0 +1,9 @@
+import torch.nn as nn
+
+class RMSLELoss(nn.Module):
+    def __init__(self):
+        super().__init__()
+        self.mse = nn.MSELoss()
+        
+    def forward(self, pred, actual):
+        return torch.sqrt(self.mse(torch.log(pred + 1), torch.log(actual + 1)))
\ No newline at end of file
-- 
GitLab