Skip to content
Snippets Groups Projects
Commit 7bb11288 authored by Luc Giffon's avatar Luc Giffon
Browse files

solve bug scoring multiclas

parent 5e50bbaa
No related branches found
No related tags found
1 merge request!3clean scripts
......@@ -98,8 +98,7 @@ class OmpForestMulticlassClassifier(OmpForest):
predictions = self.predict(X)
if metric == 'indicator':
# todo corriger bug ici
evaluation = np.abs(np.mean(np.abs(np.sign(predictions) - y) - 1))
evaluation = np.sum(np.ones_like(predictions)[predictions == y]) / X.shape[0]
else:
raise ValueError("Unsupported metric '{}'.".format(metric))
......
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