diff --git a/WebServer/AIPython/main.py b/WebServer/AIPython/ai-trainer.py similarity index 95% rename from WebServer/AIPython/main.py rename to WebServer/AIPython/ai-trainer.py index 7966e716..3e2a0885 100644 --- a/WebServer/AIPython/main.py +++ b/WebServer/AIPython/ai-trainer.py @@ -35,10 +35,11 @@ dnn_linear_model = Sequential([ # Configure the model dnn_linear_model.compile( - optimizer=optimizers.Adam(learning_rate=0.1), + optimizer=optimizers.Adam(learning_rate=0.001), loss='mean_absolute_error' ) +# Show the summary before training the model dnn_linear_model.summary() # Train the model