Skip to content

Commit 8b04b9b

Browse files
authored
Update pytorch_lstm_ts.py
1 parent 3ee64d6 commit 8b04b9b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

qlib/contrib/model/pytorch_lstm_ts.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def metric_fn(self, pred, label):
154154
mask = torch.isfinite(label)
155155

156156
if self.metric in ("", "loss"):
157-
return -self.loss_fn(pred[mask], label[mask], weight = None)
157+
return -self.loss_fn(pred[mask], label[mask], weight=None)
158158

159159
raise ValueError("unknown metric `%s`" % self.metric)
160160

0 commit comments

Comments
 (0)