From 357ee74b6fb714759f0751ea8cb40c185feefa33 Mon Sep 17 00:00:00 2001 From: you-n-g Date: Mon, 7 Feb 2022 00:05:49 +0800 Subject: [PATCH] Update pytorch_lstm_ts.py --- qlib/contrib/model/pytorch_lstm_ts.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qlib/contrib/model/pytorch_lstm_ts.py b/qlib/contrib/model/pytorch_lstm_ts.py index 70b8b0ce8..e703130fb 100755 --- a/qlib/contrib/model/pytorch_lstm_ts.py +++ b/qlib/contrib/model/pytorch_lstm_ts.py @@ -154,7 +154,7 @@ class LSTM(Model): mask = torch.isfinite(label) 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)