Bugfix: Linear regression: Use of correct method for verifying a coun…

…terfactual
andreArtelt · May 12, 2020 · e3f37d5 · e3f37d5
1 parent 7769f20
commit e3f37d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ceml/sklearn/linearregression.py b/ceml/sklearn/linearregression.py
@@ -133,7 +133,7 @@ def solve(self, x_orig, y_target, regularization, features_whitelist, return_as_
         if return_as_dict is True:
             return self._SklearnCounterfactual__build_result_dict(xcf, y_target, delta)
         else:
-            return xcf, y_target, delta
+            return xcf, self._model_predict([xcf]), delta
 
 
 def linearregression_generate_counterfactual(model, x, y_target, features_whitelist=None, regularization="l1", C=1.0, optimizer="mp", return_as_dict=True, done=None):