Added loss type to solver errors

aldro61 · Jun 2, 2017 · c3d1ba5 · c3d1ba5
1 parent a1ebd1b
commit c3d1ba5
Showing 1 changed file with 6 additions and 4 deletions.
diff --git 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]