Update pytorch_alstm.py

microsoft · May 21, 2024 · 7d77d54 · 7d77d54
1 parent af15d97
commit 7d77d54
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qlib/contrib/model/pytorch_alstm.py b/qlib/contrib/model/pytorch_alstm.py
@@ -148,7 +148,7 @@ def loss_fn(self, pred, label):
     def metric_fn(self, pred, label):
         mask = torch.isfinite(label)
 
-        if self.metric in ("", "loss"):
+        if self.metric in ("", "loss","mse"):
             return -self.loss_fn(pred[mask], label[mask])
 
         raise ValueError("unknown metric `%s`" % self.metric)