Provide input shape to tensor flow model as tuple.

linqs · Mar 15, 2024 · 30e012f · 30e012f
1 parent bc56d1e
commit 30e012f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py b/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py
@@ -34,7 +34,7 @@ def __init__(self):
 
     def internal_init(self, application, options = {}):
         layers = [
-            tensorflow.keras.layers.Input(int(options['input_shape'])),
+            tensorflow.keras.layers.Input((int(options['input_shape']), )),
             tensorflow.keras.layers.Dense(int(options['output_shape']), activation = 'softmax'),
         ]