Update 'train.py'

This commit is contained in:
Wojciech Mikołajski 2023-06-17 20:42:44 +02:00
parent 315ab5a573
commit 9b1ea21126

View File

@ -49,5 +49,5 @@ model.compile(
)
model.fit(X_train_scaled, y_train, batch_size=32, epochs=5, verbose=2)
model.save("app/savedmodel")
model.save("savedmodel")