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

Author: [zebinyang]

simtree/simtree.py

@@ -156,15 +156,17 @@ def get_projection_equation(self, node_id, precision=3):
         sortind = np.argsort(importance)[::-1]
         for i in range(est.beta_.shape[0]):
             if i == 0:
-                equation += str(round(np.abs(est.beta_[sortind[i], 0]), 3)) + self.feature_names[sortind[i]]
+                if est.beta_[sortind[i], 0] < 0:
+                    equation += " - "
+                equation += str(round(np.abs(est.beta_[sortind[i], 0]), precision)) + self.feature_names[sortind[i]]
                 continue
             else:
                 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]]
+                    equation += str(round(np.abs(est.beta_[sortind[i], 0]), precision)) + self.feature_names[sortind[i]]
         return equation
 
     def get_sparsity(self, node_id):