Commit 31a58a03 authored by Léo Bouscarrat's avatar Léo Bouscarrat
Browse files

Added correction for edge case pred =

parent 0b0af680
......@@ -109,6 +109,7 @@ class Trainer(object):
y_pred = model.predict_no_weights(X)
if type(model) is OmpForestBinaryClassifier:
y_pred = np.sign(y_pred)
y_pred = np.where(y_pred==0, 1, y_pred)
result = self._classification_score_metric(y_true, y_pred)
return result
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment