removed unneeded arguments

Meatssauce · Sep 13, 2021 · 901de26 · 901de26
1 parent f317143
commit 901de26
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/SCINet.py b/SCINet.py
@@ -132,7 +132,7 @@ def call(self, inputs):
 
 
 class StackedSciNet(tf.keras.layers.Layer):
-    def __init__(self, horizon: int, features: int, stacks: int, levels: int, h: int, kernel_size: int,
+    def __init__(self, horizon: int, stacks: int, levels: int, h: int, kernel_size: int,
                  regularizer: Tuple[float, float] = (0, 0), **kwargs):
         """
         :param horizon: number of time stamps in output

diff --git a/main.py b/main.py
@@ -20,7 +20,7 @@ def make_model(input_shape, output_shape):
     # y = SciNet(forecast_horizon, levels=levels, h=h, kernel_size=kernel_size, regularizer=(0.001, 0.01))(x)
     # model = tf.keras.Model(x, y)
     targets = tf.keras.Input(shape=(output_shape[1]), name='targets')
-    predictions = StackedSciNet(horizon, input_shape[2], stacks=K, levels=L, h=h, kernel_size=kernel_size,
+    predictions = StackedSciNet(horizon, stacks=K, levels=L, h=h, kernel_size=kernel_size,
                                 regularizer=(0.001, 0.01))(inputs, targets)
     model = tf.keras.Model(inputs=[inputs, targets], outputs=predictions)