From 7f8b87696589d820054e014b632afe53dbc403d9 Mon Sep 17 00:00:00 2001
From: jakubknczny <jakubknczny@gmail.com>
Date: Sat, 15 May 2021 21:10:00 +0200
Subject: [PATCH] try InputLayer

---
 lab5/train/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lab5/train/train.py b/lab5/train/train.py
index a495381..c3c3c65 100644
--- 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.InputLayer(shape=(12,)),
+                          layers.InputLayer(input_shape=(12,)),
                           layers.Dense(32),
                           layers.Dense(16),
                           layers.Dense(2, activation='softmax')