Merge pull request #10 from jettify/fix-regression-printer

Fix regression printer

README.rst

@@ -55,6 +55,7 @@ Simple example
 Please check full Titanic example here: https://github.com/jettify/ibreakdown/blob/master/examples/titanic.py
 
 .. code::
+
    +------------------------------------+-----------------+--------------------+--------------------+
    | Feature Name                       | Feature Value   |   Contrib:Deceased |   Contrib:Survived |
    +------------------------------------+-----------------+--------------------+--------------------|

ibreakdown/explainer.py

@@ -2,11 +2,11 @@
 
 from .explanation import ClassificationExplanation, RegressionExplanation
 from .utils import (
+    feature_group_values,
+    features_groups,
+    magnituge,
     normalize_array,
     to_matrix,
-    magnituge,
-    features_groups,
-    feature_group_values,
 )
 
 

ibreakdown/explanation.py

@@ -41,7 +41,9 @@ def _build_df(self):
         ]
         feature_names = ['intercept'] + feature_names + ['PREDICTION']
         feature_values = [None] + self.feature_values + [None]
-        contrib = [self.intercept] + self.contributions + [self.pred_value]
+        contrib = (
+            [self.intercept] + self.contributions.tolist() + [self.pred_value]
+        )
 
         data = {
             'Feature Name': feature_names,

tests/test_explainer.py

@@ -1,5 +1,8 @@
+import io
+
 import pytest
 import numpy as np
+
 from sklearn.datasets import load_boston
 from sklearn.datasets import load_iris
 from sklearn.ensemble import RandomForestRegressor, RandomForestClassifier
@@ -30,6 +33,9 @@ def test_regression(seed):
         assert np.sum(exp.contributions) + exp.intercept == pytest.approx(
             pred[0]
         )
+        with io.StringIO() as buf:
+            exp.print(file=buf, flush=True)
+            assert len(buf.getvalue()) > 0
 
 
 def test_multiclass(seed):
@@ -53,3 +59,7 @@ def test_multiclass(seed):
         # check invariant
         invariant = np.sum(exp.contributions, axis=0) + exp.intercept
         assert invariant == pytest.approx(pred[0])
+
+        with io.StringIO() as buf:
+            exp.print(file=buf, flush=True)
+            assert len(buf.getvalue()) > 0