fixing bugs

GTA-UFRJ · Jun 26, 2020 · bab4d93 · bab4d93
1 parent 1e5ea39
commit bab4d93
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dfedForest.py b/dfedForest.py
@@ -71,7 +71,7 @@ def createTree(self,depth):
 
     def createRandomTree(self,depth):
         self.newRandomTree = RandomForestClassifier(n_estimators=1,max_depth=depth)
-        self.newRandomTree = self.newTree.fit(self.data, self.label)
+        self.newRandomTree = self.newRandomTree.fit(self.data, self.label)
 
     # Show all tree on the list
     def getTrees(self):