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

Author: ZebinYang

examples/demo.ipynb

@@ -5,8 +5,8 @@
    "execution_count": 1,
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2021-12-27T12:22:23.721251Z",
-     "start_time": "2021-12-27T12:22:22.456359Z"
+     "end_time": "2021-12-27T12:56:44.118235Z",
+     "start_time": "2021-12-27T12:56:42.884995Z"
     }
    },
    "outputs": [],
@@ -32,8 +32,8 @@
    "execution_count": 2,
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2021-12-27T12:22:23.745780Z",
-     "start_time": "2021-12-27T12:22:23.722805Z"
+     "end_time": "2021-12-27T12:56:44.142989Z",
+     "start_time": "2021-12-27T12:56:44.119690Z"
     }
    },
    "outputs": [],
@@ -48,8 +48,8 @@
    "execution_count": 3,
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2021-12-27T12:00:04.703220Z",
-     "start_time": "2021-12-27T11:58:43.105225Z"
+     "end_time": "2021-12-27T12:58:05.418585Z",
+     "start_time": "2021-12-27T12:56:44.143926Z"
     },
     "scrolled": true
    },
@@ -94,8 +94,8 @@
    "execution_count": 4,
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2021-12-27T12:00:07.483934Z",
-     "start_time": "2021-12-27T12:00:04.704162Z"
+     "end_time": "2021-12-27T12:58:08.216658Z",
+     "start_time": "2021-12-27T12:58:05.419872Z"
     }
    },
    "outputs": [
@@ -128,11 +128,11 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 5,
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2021-12-27T12:22:28.522606Z",
-     "start_time": "2021-12-27T12:22:28.338619Z"
+     "end_time": "2021-12-27T12:58:08.449903Z",
+     "start_time": "2021-12-27T12:58:08.217997Z"
     }
    },
    "outputs": [],
@@ -147,8 +147,8 @@
    "execution_count": 6,
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2021-12-27T12:04:03.092316Z",
-     "start_time": "2021-12-27T12:00:32.684692Z"
+     "end_time": "2021-12-27T13:01:41.797837Z",
+     "start_time": "2021-12-27T12:58:08.450917Z"
     }
    },
    "outputs": [
@@ -192,8 +192,8 @@
    "execution_count": 7,
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2021-12-27T12:05:37.162885Z",
-     "start_time": "2021-12-27T12:04:03.093335Z"
+     "end_time": "2021-12-27T13:03:16.196360Z",
+     "start_time": "2021-12-27T13:01:41.798966Z"
     }
    },
    "outputs": [

simtree/customtree.py

@@ -43,16 +43,16 @@ def build_root(self):
     def build_leaf(self, sample_indice):
 
         if len(self.param_dict) == 0:
-            self.base_estimator.fit(self.x[sample_indice], self.y[sample_indice].ravel())
-            best_estimator = self.base_estimator
+            best_estimator = clone(self.base_estimator)
+            best_estimator.fit(self.x[sample_indice], self.y[sample_indice].ravel())
         else:
             param_size = 1
             for key, item in self.param_dict.items():
                 param_size *= len(item)
             if param_size == 1:
-                self.base_estimator.set_params(**{key: item[0] for key, item in self.param_dict.items()})
-                self.base_estimator.fit(self.x[sample_indice], self.y[sample_indice].ravel())
-                best_estimator = self.base_estimator
+                best_estimator = clone(self.base_estimator)
+                best_estimator.set_params(**{key: item[0] for key, item in self.param_dict.items()})
+                best_estimator.fit(self.x[sample_indice], self.y[sample_indice].ravel())
             else:
                 grid = GridSearchCV(self.base_estimator, param_grid=self.param_dict,
                               scoring={"mse": make_scorer(mean_squared_error, greater_is_better=False)},