wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-08-16 11:26:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

formatting

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-06-18 22:14:30 +02:00
parent bab3716819
commit ac1e89e2cb
5 changed files with 11 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/model/test_deepvar.py
+6 -31
View File
@@ -46,28 +46,13 @@ metadata = dataset.metadata
estimator = DeepVAREstimator
#@pytest.mark.timeout(10)
# @pytest.mark.timeout(10)
@pytest.mark.parametrize(
"distr_output, num_batches_per_epoch, Estimator, use_marginal_transformation",
[
(
NormalOutput(dim=target_dim),
10,
estimator,
True,
),
(
NormalOutput(dim=target_dim),
10,
estimator,
False,
),
(
LowRankMultivariateNormalOutput(dim=target_dim, rank=2),
10,
estimator,
True,
),
(NormalOutput(dim=target_dim), 10, estimator, True,),
(NormalOutput(dim=target_dim), 10, estimator, False,),
(LowRankMultivariateNormalOutput(dim=target_dim, rank=2), 10, estimator, True,),
(
LowRankMultivariateNormalOutput(dim=target_dim, rank=2),
10,
@@ -75,18 +60,8 @@ estimator = DeepVAREstimator
False,
),
(None, 10, estimator, True),
(
MultivariateNormalOutput(dim=target_dim),
10,
estimator,
True,
),
(
MultivariateNormalOutput(dim=target_dim),
10,
estimator,
False,
),
(MultivariateNormalOutput(dim=target_dim), 10, estimator, True,),
(MultivariateNormalOutput(dim=target_dim), 10, estimator, False,),
],
)
def test_deepvar(