:/

2021-05-15 21:50:50 +02:00 · 2021-05-15 21:50:50 +02:00 · 0ecc2edb39
commit 0ecc2edb39
parent 7041cd0703
1 changed files with 1 additions and 0 deletions
--- 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')