liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-12 23:36:54 +08:00
Code Activity

Update pytorch_lstm_ts.py

Browse Source
This commit is contained in:
you-n-g
2022-02-07 00:05:49 +08:00
committed by GitHub
parent 5da5cf5175
commit 357ee74b6f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
qlib/contrib/model/pytorch_lstm_ts.py
View File

@@ -154,7 +154,7 @@ class LSTM(Model):
mask = torch.isfinite(label) mask = torch.isfinite(label)
if self.metric in ("", "loss"): if self.metric in ("", "loss"):
return -self.loss_fn(pred[mask], label[mask], weight = None) return -self.loss_fn(pred[mask], label[mask], weight=None)
raise ValueError("unknown metric `%s`" % self.metric) raise ValueError("unknown metric `%s`" % self.metric)