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

clean up

Browse Source
This commit is contained in:
Dong Zhou
2021-07-22 20:16:31 +08:00
committed by you-n-g
parent 4f9f978909
commit 5b7b48e376
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
qlib/contrib/data/dataset.py
View File

@@ -225,8 +225,6 @@ class MTSDatasetH(DatasetH):
raise ValueError("cannot assign data as `num_states==0`") raise ValueError("cannot assign data as `num_states==0`")
if isinstance(vals, torch.Tensor): if isinstance(vals, torch.Tensor):
vals = vals.detach().cpu().numpy() vals = vals.detach().cpu().numpy()
# if isinstance(index, pd.Series):
# index = index.index # daily batch use Series to store index
self._memory[index] = vals self._memory[index] = vals
def clear_memory(self): def clear_memory(self):
@@ -234,7 +232,6 @@ class MTSDatasetH(DatasetH):
raise ValueError("cannot clear memory as `num_states==0`") raise ValueError("cannot clear memory as `num_states==0`")
self._memory[:] = 0 self._memory[:] = 0
# TODO: better train/eval mode design
def train(self): def train(self):
"""enable traning mode""" """enable traning mode"""
self.batch_size, self.n_samples, self.drop_last, self.shuffle = self.params self.batch_size, self.n_samples, self.drop_last, self.shuffle = self.params