Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Menu
Open sidebar
Naundorf
HPC_Vehicle_Classification
Commits
8428d970
Commit
8428d970
authored
Jun 18, 2022
by
Naundorf
Browse files
Update KI-KNN
parent
940cdc07
Pipeline
#6558
failed with stage
in 2 minutes and 20 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
KI-KNN
View file @
8428d970
...
...
@@ -116,3 +116,11 @@ predict(X_test, y_test)
plot_model_accuracy
(
history
)
plot_model_loss
(
history
)
def
predict
(
X_test
,
y_test
):
print
(
"Predictions....."
)
predictions
=
np
.
argmax
(
model
.
predict
(
X_test
),
axis
=-
1
)
target_names
=
[
constants
.0
,
constants
.1
]
print
(
classification_report
(
y_test
,
predictions
,
target_names
=
target_names
))
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