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

no need for input_size its always 1

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-02-10 13:22:41 +01:00
parent 99e144fd04
commit b053285fe9
2 changed files with 0 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/simple_feedforward/simple_feedforward_estimator.py
-4
View File
@@ -77,7 +77,6 @@ class SimpleFeedForwardEstimator(PTSEstimator):
self,
freq: str,
prediction_length: int,
input_size: int,
trainer: Trainer = Trainer(),
num_hidden_dimensions: Optional[List[int]] = None,
context_length: Optional[int] = None,
@@ -96,7 +95,6 @@ class SimpleFeedForwardEstimator(PTSEstimator):
if num_hidden_dimensions is not None
else list([40, 40])
)
self.input_size = input_size
self.prediction_length = prediction_length
self.context_length = (
context_length if context_length is not None else prediction_length
@@ -136,7 +134,6 @@ class SimpleFeedForwardEstimator(PTSEstimator):
self, device: torch.device
) -> SimpleFeedForwardTrainingNetwork:
return SimpleFeedForwardTrainingNetwork(
input_size=self.input_size,
num_hidden_dimensions=self.num_hidden_dimensions,
prediction_length=self.prediction_length,
context_length=self.context_length,
@@ -154,7 +151,6 @@ class SimpleFeedForwardEstimator(PTSEstimator):
device: torch.device,
) -> PTSPredictor:
prediction_network = SimpleFeedForwardPredictionNetwork(
input_size=self.input_size,
num_hidden_dimensions=self.num_hidden_dimensions,
prediction_length=self.prediction_length,
context_length=self.context_length,
pts/model/simple_feedforward/simple_feedforward_network.py
-2
View File
@@ -38,7 +38,6 @@ class SimpleFeedForwardNetworkBase(nn.Module):
def __init__(
self,
input_size: int,
num_hidden_dimensions: List[int],
prediction_length: int,
context_length: int,
@@ -48,7 +47,6 @@ class SimpleFeedForwardNetworkBase(nn.Module):
) -> None:
super().__init__()
self.input_size = input_size
self.num_hidden_dimensions = num_hidden_dimensions
self.prediction_length = prediction_length
self.context_length = context_length