Try to fix the AI some more
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user