Spaces:
Sleeping
Sleeping
Jensen-holm
commited on
Commit
·
d04aaf5
1
Parent(s):
bff0f22
fixed bug that returned same number twice in response
Browse files- ml-vis +1 -1
- nn/train.py +0 -1
ml-vis
CHANGED
@@ -1 +1 @@
|
|
1 |
-
Subproject commit
|
|
|
1 |
+
Subproject commit bebe25b27a895c1de71743fbf808b8e592e80806
|
nn/train.py
CHANGED
@@ -86,7 +86,6 @@ def train(nn: NN) -> dict:
|
|
86 |
return {
|
87 |
"loss_hist": loss_hist,
|
88 |
"log_loss": log_loss(y_true=y_test, y_pred=y_hat),
|
89 |
-
"accuracy": accuracy_score(y_true=y_test, y_pred=y_hat),
|
90 |
"accuracy_scores": accuracy_scores,
|
91 |
"test_accuracy": accuracy_score(y_true=y_test, y_pred=y_hat)
|
92 |
}
|
|
|
86 |
return {
|
87 |
"loss_hist": loss_hist,
|
88 |
"log_loss": log_loss(y_true=y_test, y_pred=y_hat),
|
|
|
89 |
"accuracy_scores": accuracy_scores,
|
90 |
"test_accuracy": accuracy_score(y_true=y_test, y_pred=y_hat)
|
91 |
}
|