Skip to content
Snippets Groups Projects
Commit 1adfbf0a authored by Charly Lamothe's avatar Charly Lamothe
Browse files

Fix binary classif acc metric bug, and add result of breast cancer for stage 1.

parent 559d73c0
No related branches found
No related tags found
1 merge request!9Resolve "Experiment pipeline"
......@@ -96,6 +96,8 @@ class Trainer(object):
result = self._regression_score_metric(y_true, y_pred)
elif type(model) in [OmpForestBinaryClassifier, OmpForestMulticlassClassifier, RandomForestClassifier]:
y_pred = model.predict(X)
if type(model) is OmpForestBinaryClassifier:
y_pred = y_pred.round()
result = self._classification_score_metric(y_true, y_pred)
return result
......
results/breast_cancer/stage1/losses.png

49.5 KiB

0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment