Try to fix the AI some more

This commit is contained in:
2026-03-09 22:02:49 -07:00
parent a5ca197376
commit 234b9dcc70
3 changed files with 34 additions and 28 deletions
+8 -6
View File
@@ -35,12 +35,14 @@ def TrainAI():
# Create the DNN
dnn_model = Sequential([
layers.Input(shape=(train_features.shape[1],)), # Load the feature count dynamically
layers.Dense(256, activation='elu'),
layers.Dropout(0.1), # Small dropout to prevent it from memorizing noise
layers.Dense(128, activation='elu'),
layers.Dropout(0.1), # Small dropout to prevent it from memorizing noise
layers.Dense(24, activation='elu'),
layers.Dense(1, activation='linear')
layers.Dense(256, activation='elu'), # DNN layer
layers.BatchNormalization(), # Nomralize
layers.Dropout(0.1), # Small dropout to prevent it from memorizing noise
layers.Dense(128, activation='elu'), # DNN layer
layers.BatchNormalization(), # Nomralize
layers.Dropout(0.1), # Small dropout to prevent it from memorizing noise
layers.Dense(24, activation='elu'), # DNN layer
layers.Dense(1, activation='linear') # DNN layer
])
# Allow negative numbers