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

Update pytorch_localformer_ts.py

Browse Source
This commit is contained in:
Ying-Tao Luo
2021-07-18 22:37:41 +08:00
committed by you-n-g
parent 59d4bc9394
commit e006ef40ad
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -42,7 +42,7 @@ class LocalformerModel(Model):
optimizer="adam", optimizer="adam",
reg=1e-3, reg=1e-3,
n_jobs=10, n_jobs=10,
GPU=2, GPU=0,
seed=None, seed=None,
**kwargs **kwargs
): ):
@@ -153,9 +153,6 @@ class LocalformerModel(Model):
dl_train = dataset.prepare("train", col_set=["feature", "label"], data_key=DataHandlerLP.DK_L) dl_train = dataset.prepare("train", col_set=["feature", "label"], data_key=DataHandlerLP.DK_L)
dl_valid = dataset.prepare("valid", col_set=["feature", "label"], data_key=DataHandlerLP.DK_L) dl_valid = dataset.prepare("valid", col_set=["feature", "label"], data_key=DataHandlerLP.DK_L)
import pdb
pdb.set_trace()
dl_train.config(fillna_type="ffill+bfill") # process nan brought by dataloader dl_train.config(fillna_type="ffill+bfill") # process nan brought by dataloader
dl_valid.config(fillna_type="ffill+bfill") # process nan brought by dataloader dl_valid.config(fillna_type="ffill+bfill") # process nan brought by dataloader