From b8961d4f66e0a19ecf6b6e3b307df3f8212eeca1 Mon Sep 17 00:00:00 2001
From: jakubknczny <jakubknczny@gmail.com>
Date: Sat, 15 May 2021 20:52:45 +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 0f3c519..a495381 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.Input(shape=(12,)),
+                          layers.InputLayer(shape=(12,)),
                           layers.Dense(32),
                           layers.Dense(16),
                           layers.Dense(2, activation='softmax')