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

time feature is bigger since we need it when sampling

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-02-03 16:05:46 +01:00
parent ccc1efbabd
commit f0b4fc988d
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/deepvar/deepvar_network.py
+2 -2
View File
@@ -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,
pts/model/transformer_tempflow/transformer_tempflow_network.py
+2 -2
View File
@@ -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,