Skip to content

Commit 0062511

Browse files
committed
[tsa/eval] - fix: invert_scale no longer overwrites targets
1 parent 81210d6 commit 0062511

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tsa/eval.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,11 +44,11 @@ def evaluate(test_iter, criterion, model, config, ts):
4444
predictions, targets = torch.cat(predictions), torch.cat(targets)
4545

4646
if config.general.do_eval:
47-
preds, targets = ts.invert_scale(predictions), ts.invert_scale(targets)
47+
preds, targets_ = ts.invert_scale(predictions), ts.invert_scale(targets)
4848

4949
plt.figure()
5050
plt.plot(preds, linewidth=.3)
51-
plt.plot(targets, linewidth=.3)
51+
plt.plot(targets_, linewidth=.3)
5252
plt.savefig("{}/preds.png".format(config.general.output_dir))
5353
plt.close()
5454

0 commit comments

Comments
 (0)