Fixed selector bug around regression.

interpretml · Jul 21, 2020 · 6ddc0cc · 6ddc0cc
1 parent 936bd04
commit 6ddc0cc
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/interpret-core/interpret/utils/all.py b/python/interpret-core/interpret/utils/all.py
@@ -118,9 +118,8 @@ def gen_local_selector(data_dicts, round=3, is_classification=True):
         record["PrScore"] = perf_dict["predicted_score"]
         record["AcScore"] = perf_dict["actual_score"]
 
-        if is_classification:
-            record["Predicted"] = perf_dict["predicted"]
-            record["Actual"] = perf_dict["actual"]
+        record["Predicted"] = perf_dict["predicted"]
+        record["Actual"] = perf_dict["actual"]
 
         record["Resid"] = record["AcScore"] - record["PrScore"]
         record["AbsResid"] = abs(record["Resid"])