mirror of
https://github.com/Doctorado-ML/Odte.git
synced 2025-07-11 08:12:06 +00:00
Fix nodes_leaves for base_estimator
This commit is contained in:
parent
3558c946a8
commit
74343a15e1
13
odte/Odte.py
13
odte/Odte.py
@ -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
|
||||||
|
Loading…
x
Reference in New Issue
Block a user