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

Update test_dataset

Browse Source
This commit is contained in:
lwwang1995
2020-12-05 22:39:30 +08:00
committed by you-n-g
parent 60f62482b7
commit bf8de72605
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_dataset.py
View File

@@ -62,7 +62,7 @@ class TestDataset(TestAutoData):
) )
tsds_train = tsdh.prepare("train") # Test the correctness tsds_train = tsdh.prepare("train") # Test the correctness
tsds = tsdh.prepare("valid") # prepare a dataset with is friendly to converting tabular data to time-series tsds = tsdh.prepare("valid") # prepare a dataset with is friendly to converting tabular data to time-series
train_loader = DataLoader(tsds_train, batch_size=800, shuffle=True) train_loader = DataLoader(tsds_train, batch_size=800, shuffle=True, num_workers=10)
for data in train_loader: for data in train_loader:
now = time.localtime() now = time.localtime()
print(time.strftime("%Y-%m-%d-%H_%M_%S", now)) print(time.strftime("%Y-%m-%d-%H_%M_%S", now))