Rename main to ai-trainer.py

2026-02-17 17:42:35 -08:00
parent eb6ccb66e6
commit 6cf70f1e37
1 changed files with 2 additions and 1 deletions
@@ -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