s452487
/
ium_452487
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Upload files to "/"

This commit is contained in:
s452487 2024-05-04 15:06:38 +02:00
parent c98af9413d
commit 89d55cd49a
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
train.py
View File

@ -52,10 +52,22 @@ model = create_model()
model.summary()
import os
import json
if 'training_parameters' not in os.environ:
parameter_epochs = 11
else:
training_parameters = os.environ.get('training_parameters')
parameter_dict = json.loads(training_parameters)
parameter_epochs = parameter_dict["epochs"]
# Early stopping dla regularyzacji
callback = keras.callbacks.EarlyStopping(monitor='val_loss', mode='min', patience=3, restore_best_weights=True)
history = model.fit(train_x, train_y, validation_data=(test_x, test_y), epochs=11, callbacks=[callback])
history = model.fit(train_x, train_y, validation_data=(test_x, test_y), epochs=parameter_epochs, callbacks=[callback])
model.save("model.keras")