Resolve "Experiment pipeline"

Merge to master the beginning of density plotting.

code/bolsonaro/trainer.py

@@ -128,10 +128,12 @@ class Trainer(object):
         """
 
         model_weights = ''
-        if type(model) == RandomForestRegressor:
-            model_weights = model.coef_
-        elif type(model) == OmpForestRegressor:
+        if type(model) in [OmpForestRegressor, OmpForestBinaryClassifier]:
             model_weights = model._omp.coef_
+        elif type(model) == OmpForestMulticlassClassifier:
+            model_weights = model._dct_class_omp
+        elif type(model) == OmpForestBinaryClassifier:
+            model_weights = model._omp
 
         results = ModelRawResults(
             model_weights=model_weights,