for a fair comparison, adjust roughness by map x to be within [0, 1]; version 0.1.8

Author: [zebinyang]

simtree/simtree.py

@@ -159,11 +159,12 @@ def get_projection_equation(self, node_id, precision=3):
                 equation += str(round(np.abs(est.beta_[sortind[i], 0]), 3)) + self.feature_names[sortind[i]]
                 continue
             else:
-                if est.beta_[sortind[i], 0] > 0:
-                    equation += " + "
-                else:
-                    equation += " - "
-                equation += str(round(np.abs(est.beta_[sortind[i], 0]), 3)) + self.feature_names[sortind[i]]
+                if np.abs(est.beta_[sortind[i], 0]) > 0:
+                    if est.beta_[sortind[i], 0] > 0:
+                        equation += " + "
+                    else:
+                        equation += " - "
+                    equation += str(round(np.abs(est.beta_[sortind[i], 0]), 3)) + self.feature_names[sortind[i]]
         return equation
 
     def get_sparsity(self, node_id):