Merge branch 'master' of https://github.com/csinva/imodels

Author: csinva

imodels/tree/figs.py

@@ -378,7 +378,10 @@ def _importances(node: Node):
             # require the tree to have more than 1 node, otherwise just leave importance_data_tree as zeros
             if 1 < next(node_counter):
                 tree_samples = _importances(tree_)
-                importance_data_tree /= tree_samples
+                if tree_samples != 0:
+                    importance_data_tree /= tree_samples
+                else:
+                    importance_data_tree = 0
 
             importance_data.append(importance_data_tree)
 

imodels/tree/hierarchical_shrinkage.py

@@ -183,11 +183,21 @@ def __repr__(self):
 
 
 class HSTreeRegressor(HSTree, RegressorMixin):
-    ...
+    def __init__(self, estimator_: BaseEstimator = DecisionTreeRegressor(max_leaf_nodes=20),
+                 reg_param: float = 1, shrinkage_scheme_: str = 'node_based'):
+        super().__init__(estimator_=estimator_,
+                         reg_param=reg_param,
+                         shrinkage_scheme_=shrinkage_scheme_,
+                         )
 
 
 class HSTreeClassifier(HSTree, ClassifierMixin):
-    ...
+    def __init__(self, estimator_: BaseEstimator = DecisionTreeClassifier(max_leaf_nodes=20),
+                 reg_param: float = 1, shrinkage_scheme_: str = 'node_based'):
+        super().__init__(estimator_=estimator_,
+                         reg_param=reg_param,
+                         shrinkage_scheme_=shrinkage_scheme_,
+                         )
 
 
 def _get_cv_criterion(scorer):