From 7041cd0703e3a4745554d87a88eb85a8822bc3b1 Mon Sep 17 00:00:00 2001
From: jakubknczny <jakubknczny@gmail.com>
Date: Sat, 15 May 2021 21:35:32 +0200
Subject: [PATCH] adjust model

---
 lab5/eval/requirements.txt | 14 ++++----------
 lab5/train/train.py        |  1 -
 2 files changed, 4 insertions(+), 11 deletions(-)

diff --git a/lab5/eval/requirements.txt b/lab5/eval/requirements.txt
index e5d5ae2..33e4e92 100644
--- a/lab5/eval/requirements.txt
+++ b/lab5/eval/requirements.txt
@@ -1,10 +1,4 @@
-h5py==2.10.0
-Keras==2.4.3
-Keras-Preprocessing==1.1.2
-keras-vis==0.4.1
-numpy==1.19.5
-pandas==1.1.5
-pandas-datareader==0.9.0
-pandas-gbq==0.13.3
-pandas-profiling==1.4.1
-tensorflow==2.4.1
\ No newline at end of file
+numpy~=1.19.2
+pandas
+tensorflow
+keras==2.4.3
diff --git a/lab5/train/train.py b/lab5/train/train.py
index c3c3c65..e3f398a 100644
--- a/lab5/train/train.py
+++ b/lab5/train/train.py
@@ -17,7 +17,6 @@ Y_valid = X_valid.pop('stabf')
 Y_valid = pd.get_dummies(Y_valid)
 
 model = tensorflow.keras.Sequential([
-                          layers.InputLayer(input_shape=(12,)),
                           layers.Dense(32),
                           layers.Dense(16),
                           layers.Dense(2, activation='softmax')