1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-01 18:11:18 +08:00
This commit is contained in:
Dong Zhou
2021-07-22 20:16:31 +08:00
committed by you-n-g
parent 4f9f978909
commit 5b7b48e376

View File

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