diff --git a/code/bolsonaro/models/omp_forest_classifier.py b/code/bolsonaro/models/omp_forest_classifier.py index 02c993c63fa72dd78d65a075068ae1b91c22ef8e..e121ac7cdb6d8957b75cde3155c0df3816ba834b 100644 --- a/code/bolsonaro/models/omp_forest_classifier.py +++ b/code/bolsonaro/models/omp_forest_classifier.py @@ -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))