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

Update models.

Browse Source
This commit is contained in:
lwwang1995
2020-12-07 14:22:34 +08:00
committed by you-n-g
parent ec0d7838ac
commit 65a9a72a88
5 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
qlib/contrib/model/pytorch_alstm_ts.py
View File

@@ -262,7 +262,7 @@ class ALSTM(Model):
dl_test = dataset.prepare("test", data_key=DataHandlerLP.DK_I)
dl_test.config(fillna_type="ffill+bfill")
test_loader = DataLoader(dl_test, batch_size=self.batch_size, num_workers=20)
test_loader = DataLoader(dl_test, batch_size=self.batch_size, num_workers=self.n_jobs)
self.ALSTM_model.eval()
preds = []

2
qlib/contrib/model/pytorch_gats_ts.py
View File

@@ -307,7 +307,7 @@ class GATs(Model):
dl_test = dataset.prepare("test", data_key=DataHandlerLP.DK_I)
dl_test.config(fillna_type="ffill+bfill")
test_loader = DataLoader(dl_test, batch_size=self.batch_size, num_workers=20)
test_loader = DataLoader(dl_test, batch_size=self.batch_size, num_workers=self.n_jobs)
self.ALSTM_model.eval()
preds = []

2
qlib/contrib/model/pytorch_gru_ts.py
View File

@@ -262,7 +262,7 @@ class GRU(Model):
dl_test = dataset.prepare("test", data_key=DataHandlerLP.DK_I)
dl_test.config(fillna_type="ffill+bfill")
test_loader = DataLoader(dl_test, batch_size=self.batch_size, num_workers=20)
test_loader = DataLoader(dl_test, batch_size=self.batch_size, num_workers=self.n_jobs)
self.GRU_model.eval()
preds = []

2
qlib/contrib/model/pytorch_lstm_ts.py
View File

@@ -262,7 +262,7 @@ class LSTM(Model):
dl_test = dataset.prepare("test", data_key=DataHandlerLP.DK_I)
dl_test.config(fillna_type="ffill+bfill")
test_loader = DataLoader(dl_test, batch_size=self.batch_size, num_workers=20)
test_loader = DataLoader(dl_test, batch_size=self.batch_size, num_workers=self.n_jobs)
self.LSTM_model.eval()
preds = []

2
qlib/contrib/model/pytorch_sfm_ts.py
View File

@@ -433,7 +433,7 @@ class SFM(Model):
dl_test = dataset.prepare("test", data_key=DataHandlerLP.DK_I)
dl_test.config(fillna_type="ffill+bfill")
test_loader = DataLoader(dl_test, batch_size=self.batch_size, num_workers=20)
test_loader = DataLoader(dl_test, batch_size=self.batch_size, num_workers=self.n_jobs)
self.sfm_model.eval()
preds = []