From f0b4fc988d96b7a1ff41b254a5b9db73bdeae12d Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Mon, 3 Feb 2020 16:05:46 +0100 Subject: [PATCH] time feature is bigger since we need it when sampling --- pts/model/deepvar/deepvar_network.py | 4 ++-- .../transformer_tempflow/transformer_tempflow_network.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/pts/model/deepvar/deepvar_network.py b/pts/model/deepvar/deepvar_network.py index 6116058..ab20c0d 100644 --- a/pts/model/deepvar/deepvar_network.py +++ b/pts/model/deepvar/deepvar_network.py @@ -239,7 +239,7 @@ class DeepVARTrainingNetwork(nn.Module): if future_time_feat is None or future_target_cdf is None: time_feat = past_time_feat[ - :, self.history_length - self.context_length :, ... + :,- self.context_length :, ... ] sequence = past_target_cdf sequence_length = self.history_length @@ -247,7 +247,7 @@ class DeepVARTrainingNetwork(nn.Module): else: time_feat = torch.cat( ( - past_time_feat[:, self.history_length - self.context_length :, ...], + past_time_feat[:,- self.context_length :, ...], future_time_feat, ), dim=1, diff --git a/pts/model/transformer_tempflow/transformer_tempflow_network.py b/pts/model/transformer_tempflow/transformer_tempflow_network.py index 493abe7..59ee7f0 100644 --- a/pts/model/transformer_tempflow/transformer_tempflow_network.py +++ b/pts/model/transformer_tempflow/transformer_tempflow_network.py @@ -199,7 +199,7 @@ class TransformerTempFlowTrainingNetwork(nn.Module): if future_time_feat is None or future_target_cdf is None: time_feat = past_time_feat[ - :, self.history_length - self.context_length :, ... + :, - self.context_length :, ... ] sequence = past_target_cdf sequence_length = self.history_length @@ -207,7 +207,7 @@ class TransformerTempFlowTrainingNetwork(nn.Module): else: time_feat = torch.cat( ( - past_time_feat[:, self.history_length - self.context_length :, ...], + past_time_feat[:, - self.context_length :, ...], future_time_feat, ), dim=1,