Resolve "Adding new datasets"

Closes #17 (closed)

code/bolsonaro/models/omp_forest_classifier.py

@@ -48,7 +48,7 @@ class OmpForestBinaryClassifier(SingleOmpForest):
 
         result_omp = np.mean(omp_trees_predictions, axis=1)
 
-        result_omp = np.array([-1 if pred < 0.5 else 1 for pred in result_omp])
+        result_omp = (result_omp - 0.5) * 2
 
         return result_omp