From c3d1ba54af1c9216ae1d7862608a85b99e4729a8 Mon Sep 17 00:00:00 2001 From: Alexandre Drouin Date: Fri, 2 Jun 2017 15:37:06 -0400 Subject: [PATCH] Added loss type to solver errors --- mmit/learning.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/mmit/learning.py b/mmit/learning.py index 882fb87..2e4dacc 100644 --- a/mmit/learning.py +++ b/mmit/learning.py @@ -156,13 +156,15 @@ def _optimal_split(node): ----- margin: {4:.9f} -lower_bounds: {5!s} +loss type: {5!s} -upper_bounds: {6!s} +lower_bounds: {6!s} + +upper_bounds: {7!s} END -""".format(left_costs[-1], right_costs[-1], left_preds[-1], right_preds[-1], self.margin, lower_sorted.tolist(), - upper_sorted.tolist())) +""".format(left_costs[-1], right_costs[-1], left_preds[-1], right_preds[-1], self.margin, self.loss, + lower_sorted.tolist(), upper_sorted.tolist())) # Combine the values of duplicate feature values and remove splits where all examples are in one leaf unique_left_preds = left_preds[last_idx_by_value][:-1]