wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-08-19 12:30:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

set shuffling of time series in file and list dataset to false by default

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-03-09 10:51:04 +01:00
parent 6bc8de7edb
commit 37fdb0e5b7
7 changed files with 23 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/dataset/test_multivariate_grouper.py
+4 -7
View File
@@ -67,10 +67,8 @@ TRAIN_FILL_RULE = [np.mean, np.mean, np.mean, np.mean, lambda x: 0.0]
def test_multivariate_grouper_train(
univariate_ts, multivariate_ts, train_fill_rule
) -> None:
univariate_ds = ListDataset(univariate_ts, freq="1D", is_train=False)
multivariate_ds = ListDataset(
multivariate_ts, freq="1D", one_dim_target=False, is_train=False
)
univariate_ds = ListDataset(univariate_ts, freq="1D")
multivariate_ds = ListDataset(multivariate_ts, freq="1D", one_dim_target=False)
grouper = MultivariateGrouper(train_fill_rule=train_fill_rule)
assert (
@@ -117,10 +115,9 @@ MAX_TARGET_DIM = [2, 1]
def test_multivariate_grouper_test(
univariate_ts, multivariate_ts, test_fill_rule, max_target_dim
) -> None:
univariate_ds = ListDataset(univariate_ts, freq="1D", is_train=False)
univariate_ds = ListDataset(univariate_ts, freq="1D")
multivariate_ds = ListDataset(
multivariate_ts, freq="1D", one_dim_target=False, is_train=False
)
multivariate_ts, freq="1D", one_dim_target=False)
grouper = MultivariateGrouper(
test_fill_rule=test_fill_rule, num_test_dates=2, max_target_dim=max_target_dim,