Fix for intercept on regression.

interpretml · Apr 2, 2020 · 38254e7 · 38254e7
1 parent 71f572b
commit 38254e7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/interpret-core/interpret/glassbox/ebm/ebm.py b/python/interpret-core/interpret/glassbox/ebm/ebm.py
@@ -1280,8 +1280,8 @@ def explain_local(self, X, y=None, name=None):
 
         n_rows = instances.shape[1]
         data_dicts = []
+        intercept = self.intercept_
         if self.n_classes_ <= 2:
-            intercept = self.intercept_
             if isinstance(self.intercept_, np.ndarray) or isinstance(self.intercept_, list):
                 intercept = intercept[0]