From bf98d0327708d6f9e8565418957f544355cde0b6 Mon Sep 17 00:00:00 2001 From: loldebyte <47870314+loldebyte@users.noreply.github.com> Date: Mon, 17 Aug 2020 15:46:05 +0200 Subject: [PATCH] Update Random Forests.ipynb --- Random Forests.ipynb | 15 +-------------- 1 file changed, 1 insertion(+), 14 deletions(-) diff --git a/Random Forests.ipynb b/Random Forests.ipynb index c4a7564..66732d8 100644 --- a/Random Forests.ipynb +++ b/Random Forests.ipynb @@ -258,20 +258,7 @@ " X.reset_index(drop=True, inplace=True)\n", " y.reset_index(drop=True, inplace=True)\n", " \n", - " all_tree = []\n", - "\n", - " for i in range(self.n_estimators):\n", - " X_boot, y_boot = self.__bootstrap__(X, y)\n", - "\n", - " X_boot.reset_index(drop=True, inplace=True)\n", - " y_boot.reset_index(drop=True, inplace=True)\n", - "\n", - " clf = DecisionTreeClassif(max_features=self.max_features)\n", - " clf.fit(X_boot, y_boot, target_name=target_name)\n", - "\n", - " all_tree.append(clf)\n", - "\n", - " self.__all_tree__ = all_tree\n", + " self.__all_tree__ = generate_forest(X, y, self.n_estimator)\n", " \n", " def __bootstrap__(self, X, y):\n", " indices = np.random.randint(len(y), size=len(y))\n",