Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Menu
Open sidebar
fork_HPC_Vehicle_Classification
HPC_Vehicle_Classification
Commits
69e6cb7b
Commit
69e6cb7b
authored
Jun 17, 2021
by
Sintal
Browse files
Update main.py
parent
51b56bf0
Pipeline
#4396
failed with stage
in 59 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
main.py
View file @
69e6cb7b
...
...
@@ -114,7 +114,7 @@ def train_and_save_model(model, X_train, y_train, X_test, y_test):
print
(
"Start Training..."
)
history
=
model
.
fit
(
X_train
,
y_train
,
epochs
=
35
,
validation_data
=
(
X_test
,
y_test
),
callbacks
=
[
tensorboard_callback
])
# Saving the trained model to avoid re-training
model
.
save
(
constants
.
TRAINED_MODEL
)
#
model.save(constants.TRAINED_MODEL)
return
history
def
model_predict
(
model
,
X_test
,
y_test
):
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment