skip grid search cv in build leaf if param_dict is empty; version 0.2.4

Author: ZebinYang

examples/demo.ipynb

@@ -44,7 +44,7 @@ def build_leaf(self, sample_indice):
 
         if len(self.reg_lambda) > 1:
             best_estimator = LassoCV(alphas=self.reg_lambda, cv=5, precompute=False, random_state=self.random_state)
-            best_estimator.fit(nx, self.y[sample_indice], self.sample_weight[sample_indice])
+            best_estimator.fit(nx, self.y[sample_indice])
         else:
             if self.reg_lambda[0] > 0:
                 best_estimator = Lasso(alpha=self.reg_lambda[0], precompute=False, random_state=self.random_state)

simtree/glmtree.py

@@ -627,8 +627,6 @@ def get_loss(self, label, pred):
             containing the input dataset
         pred : array-like of shape (n_samples,)
             containing the output dataset
-        sample_weight : array-like of shape (n_samples,), optional
-            containing sample weights
         Returns
         -------
         float