This commit is contained in:
jakubknczny 2021-05-15 19:17:28 +02:00
parent db7a59f689
commit 7133025403

View File

@ -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')