s470607/ium_470607
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

:c

Browse Source
This commit is contained in:
jakubknczny 2021-05-15 19:17:28 +02:00
parent db7a59f689
commit 7133025403
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lab5/train/train.py
View File

@ -30,7 +30,7 @@ model.compile(
metrics=[tensorflow.keras.metrics.BinaryAccuracy()]) metrics=[tensorflow.keras.metrics.BinaryAccuracy()])
history = model.fit(tensorflow.convert_to_tensor(X_train, np.float32), # history = model.fit(tensorflow.convert_to_tensor(X_train, np.float32),
Y_train, epochs=2, validation_data=(X_valid, Y_valid)) history = model.fit(X_train, Y_train, epochs=2, validation_data=(X_valid, Y_valid))
model.save('grid-stability-dense.h5') model.save('grid-stability-dense.h5')