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

Author: [zebinyang]

simtree/__init__.py

@@ -1,12 +1,6 @@
-from .cart import CARTRegressor, CARTClassifier
 from .glmtree import GLMTreeRegressor, GLMTreeClassifier
-from .simtree import SIMTreeRegressor, SIMTreeClassifier
-from .customtree import CustomMobTreeRegressor, CustomMobTreeClassifier
 
-__all__ = ["CARTRegressor", "CARTClassifier",
-        "GLMTreeRegressor", "GLMTreeClassifier",
-        "SIMTreeRegressor", "SIMTreeClassifier",
-        "CustomMobTreeRegressor", "CustomMobTreeClassifier"]
+__all__ = ["GLMTreeRegressor", "GLMTreeClassifier"]
 
-__version__ = '0.1.8'
+__version__ = '0.1'
 __author__ = 'Zebin Yang'

simtree/simtree.py

@@ -126,7 +126,7 @@ def get_feature_importance(self, node_id):
             print("This is a constant node, and SIM is not available.")
             return
 
-        importance = (self.x[self.decision_path_indice(self.x, node_id)] * self.leaf_estimators_[node_id].beta_.ravel()).std(0)
+        importance = (self.x[self.decision_path_indice(self.x, node_id)] * self.leaf_estimators_[node_id].beta_.ravel()).std(0, ddof=1)
         return importance
 
     def get_projection_equation(self, node_id, precision=3):