s487187/ium_487187
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

s

Browse Source
This commit is contained in:
Jakub Zaręba 2023-05-14 15:13:13 +02:00
parent e1ca9046a6
commit 156c19a688
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -72,7 +72,7 @@ def train_model(data_file, model_file, epochs, batch_size, test_size, random_sta
print('Test accuracy:', accuracy) print('Test accuracy:', accuracy)
print('Test loss:', loss) print('Test loss:', loss)
model.save(model_file) model.save("model.h5")
input_signature = { input_signature = {
'input': tensor_spec.TensorSpec(shape=X_train[0].shape, dtype=X_train.dtype) 'input': tensor_spec.TensorSpec(shape=X_train[0].shape, dtype=X_train.dtype)
} }