Rename main to ai-trainer.py
This commit is contained in:
@@ -35,10 +35,11 @@ dnn_linear_model = Sequential([
|
|||||||
|
|
||||||
# Configure the model
|
# Configure the model
|
||||||
dnn_linear_model.compile(
|
dnn_linear_model.compile(
|
||||||
optimizer=optimizers.Adam(learning_rate=0.1),
|
optimizer=optimizers.Adam(learning_rate=0.001),
|
||||||
loss='mean_absolute_error'
|
loss='mean_absolute_error'
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Show the summary before training the model
|
||||||
dnn_linear_model.summary()
|
dnn_linear_model.summary()
|
||||||
|
|
||||||
# Train the model
|
# Train the model
|
||||||
Reference in New Issue
Block a user