Resolve "Adding new datasets"

Closes #17 (closed)

code/bolsonaro/trainer.py

@@ -108,7 +108,8 @@ class Trainer(object):
             else:
                 y_pred = model.predict_no_weights(X)
             if type(model) is OmpForestBinaryClassifier:
-                y_pred = y_pred.round()
+                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
 
@@ -187,5 +188,3 @@ class Trainer(object):
 
             self._logger.info("Base performance on dev without weights: {}".format(results.dev_score_base))
             self._logger.info("Performance on dev: {}".format(results.dev_score))
-
-