From 71330254033f20595774a043ba4ecc6304531749 Mon Sep 17 00:00:00 2001 From: jakubknczny Date: Sat, 15 May 2021 19:17:28 +0200 Subject: [PATCH] :c --- lab5/train/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lab5/train/train.py b/lab5/train/train.py index 9f24cbb..a4780e2 100644 --- a/lab5/train/train.py +++ b/lab5/train/train.py @@ -30,7 +30,7 @@ model.compile( metrics=[tensorflow.keras.metrics.BinaryAccuracy()]) -history = model.fit(tensorflow.convert_to_tensor(X_train, np.float32), - Y_train, epochs=2, validation_data=(X_valid, Y_valid)) +# history = model.fit(tensorflow.convert_to_tensor(X_train, np.float32), +history = model.fit(X_train, Y_train, epochs=2, validation_data=(X_valid, Y_valid)) model.save('grid-stability-dense.h5')