adjust model

2021-05-15 21:35:32 +02:00 · 2021-05-15 21:35:32 +02:00 · 7041cd0703
commit 7041cd0703
parent 48f74d8b32
2 changed files with 4 additions and 11 deletions
--- 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
+numpy~=1.19.2
+pandas
+tensorflow
+keras==2.4.3
--- 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')