liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-15 08:46:56 +08:00
Code Activity

update run_all_model and black format

Browse Source
This commit is contained in:
Young
2021-07-18 12:09:57 +08:00
committed by you-n-g
parent b07e0bffb1
commit 59d4bc9394
4 changed files with 41 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
qlib/contrib/model/pytorch_localformer.py
View File

@@ -24,6 +24,7 @@ from ...model.base import Model
from ...data.dataset import DatasetH, TSDatasetH
from ...data.dataset.handler import DataHandlerLP
from torch.nn.modules.container import ModuleList
# qrun examples/benchmarks/Localformer/workflow_config_localformer_Alpha360.yaml ”
@@ -150,8 +151,8 @@ class LocalformerModel(Model):
if len(indices) - i < self.batch_size:
break
feature = torch.from_numpy(x_values[indices[i: i + self.batch_size]]).float().to(self.device)
label = torch.from_numpy(y_values[indices[i: i + self.batch_size]]).float().to(self.device)
feature = torch.from_numpy(x_values[indices[i : i + self.batch_size]]).float().to(self.device)
label = torch.from_numpy(y_values[indices[i : i + self.batch_size]]).float().to(self.device)
with torch.no_grad():
pred = self.model(feature)