Update 'train.py'

This commit is contained in:
Wojciech Mikołajski 2023-06-17 20:40:50 +02:00
parent 76cdf71383
commit 3baa3736f8

View File

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