Skip to content

Commit e584961

Browse files
committed
minor refactor
1 parent 544943e commit e584961

File tree

8 files changed

+4
-3
lines changed

8 files changed

+4
-3
lines changed

exp/exp_main.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from models.ETSformer.model import ETSformer
1+
from models import ETSformer
22
from data_provider.data_factory import data_provider
33
from exp.exp_basic import Exp_Basic
44
from utils.tools import EarlyStopping, adjust_learning_rate
@@ -210,8 +210,8 @@ def test(self, setting, data, save_vals=False):
210210
outputs = outputs.detach().cpu().numpy()
211211
batch_y = batch_y.detach().cpu().numpy()
212212

213-
pred = outputs # outputs.detach().cpu().numpy() # .squeeze()
214-
true = batch_y # batch_y.detach().cpu().numpy() # .squeeze()
213+
pred = outputs
214+
true = batch_y
215215

216216
preds.append(pred)
217217
trues.append(true)

models/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
from .etsformer.model import ETSformer
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.

0 commit comments

Comments
 (0)