fix past_observed_values

This commit is contained in:
Dr. Kashif Rasul
2020-01-09 13:46:56 +01:00
parent ce37ff0f57
commit e263bd5f5c
2 changed files with 1 additions and 2 deletions
+1 -1
View File
@@ -234,7 +234,7 @@ class DeepVARTrainingNetwork(nn.Module):
"""
past_observed_values = torch.min(
past_observed_values, past_is_pad.unsqueeze(-1)
past_observed_values, 1 - past_is_pad.unsqueeze(-1)
)
if future_time_feat is None or future_target_cdf is None:
-1
View File
@@ -203,7 +203,6 @@ class IndependentNormalOutput(DistributionOutput):
class MultivariateNormalOutput(DistributionOutput):
def __init__(self, dim: int) -> None:
self.args_dim = {"loc": dim, "scale_tril": dim * dim}
self.distr_cls = MultivariateNormal
self.dim = dim
def domain_map(self, loc, scale):