Optumize the training time

2026-02-17 21:13:54 -08:00
parent 926003c6b4
commit 8d599be940
3 changed files with 202 additions and 0 deletions
@@ -52,6 +52,7 @@ def TrainAI():
    Training_Data = dnn_linear_model.fit( 
        train_features,
        train_labels,
+        batch_size=1024,
        epochs=100,
        # Show progress
        verbose=1,