Resolve "integration-sota"

Closes #15 (closed)

code/bolsonaro/trainer.py

@@ -155,7 +155,7 @@ class Trainer(object):
 
     def _evaluate_predictions(self, model, X, aggregation_function):
         if type(model) in [OmpForestRegressor, SimilarityForestRegressor, KMeansForestRegressor, EnsembleSelectionForestRegressor,
-            OmpForestBinaryClassifier, OmpForestMulticlassClassifier]:
+            OmpForestBinaryClassifier, OmpForestMulticlassClassifier, SimilarityForestClassifier, KMeansForestClassifier, EnsembleSelectionForestClassifier]:
             estimators = model.forest
             estimators = np.asarray(estimators)[model._omp.coef_ != 0]
         elif type(model) in [RandomForestRegressor, RandomForestClassifier]: