Doctorado-ML/odte
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/Odte.git synced 2025-07-11 08:12:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix nodes_leaves for base_estimator

Browse Source
This commit is contained in:
Ricardo Montañana Gómez 2021-11-24 10:50:19 +01:00
parent 3558c946a8
commit 74343a15e1
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
odte/Odte.py
View File

@ -88,13 +88,14 @@ class Odte(BaseEnsemble, ClassifierMixin):
return self return self
def _compute_metrics(self) -> None: def _compute_metrics(self) -> None:
tdepth = tnodes = tleaves = 0 tdepth = tnodes = tleaves = 0.0
for estimator in self.estimators_: for estimator in self.estimators_:
nodes, leaves = estimator.nodes_leaves() if hasattr(estimator, "nodes_leaves"):
depth = estimator.depth_ nodes, leaves = estimator.nodes_leaves()
tdepth += depth depth = estimator.depth_
tnodes += nodes tdepth += depth
tleaves += leaves tnodes += nodes
tleaves += leaves
self.depth_ = tdepth / self.n_estimators self.depth_ = tdepth / self.n_estimators
self.leaves_ = tleaves / self.n_estimators self.leaves_ = tleaves / self.n_estimators
self.nodes_ = tnodes / self.n_estimators self.nodes_ = tnodes / self.n_estimators