From 0ecc2edb39a9c759930ac72db09588794b634e75 Mon Sep 17 00:00:00 2001
From: jakubknczny <jakubknczny@gmail.com>
Date: Sat, 15 May 2021 21:50:50 +0200
Subject: [PATCH] :/

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

diff --git a/lab5/train/train.py b/lab5/train/train.py
index e3f398a..0f3c519 100644
--- a/lab5/train/train.py
+++ b/lab5/train/train.py
@@ -17,6 +17,7 @@ Y_valid = X_valid.pop('stabf')
 Y_valid = pd.get_dummies(Y_valid)
 
 model = tensorflow.keras.Sequential([
+                          layers.Input(shape=(12,)),
                           layers.Dense(32),
                           layers.Dense(16),
                           layers.Dense(2, activation='softmax')