From 8324f60cd46e68ce72b60a56add473f535f4d518 Mon Sep 17 00:00:00 2001 From: jakubknczny Date: Sat, 15 May 2021 22:06:36 +0200 Subject: [PATCH] model adds --- lab5/train/train.py | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/lab5/train/train.py b/lab5/train/train.py index 0f3c519..099c934 100644 --- a/lab5/train/train.py +++ b/lab5/train/train.py @@ -16,12 +16,18 @@ Y_test = pd.get_dummies(Y_test) 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') -]) +# model = tensorflow.keras.Sequential([ +# layers.Input(shape=(12,)), +# layers.Dense(32), +# layers.Dense(16), +# layers.Dense(2, activation='softmax') +# ]) + +model = tensorflow.keras.Sequential() +model.add(layers.Input(shape=(12,))) +model.add(layers.Dense(32)) +model.add(layers.Dense(16)) +model.add(layers.Dense(2, activation='softmax')) model.compile( loss=tensorflow.keras.losses.BinaryCrossentropy(),