try InputLayer

2021-05-15 20:52:45 +02:00 · 2021-05-15 20:52:45 +02:00 · b8961d4f66
commit b8961d4f66
parent ff2edcf654
1 changed files with 1 additions and 1 deletions
--- a/lab5/train/train.py
+++ b/lab5/train/train.py
@ -17,7 +17,7 @@ Y_valid = X_valid.pop('stabf')
 Y_valid = pd.get_dummies(Y_valid)

 model = tensorflow.keras.Sequential([
-                          layers.Input(shape=(12,)),
+                          layers.InputLayer(shape=(12,)),
                          layers.Dense(32),
                          layers.Dense(16),
                          layers.Dense(2, activation='softmax')