From 74343a15e17723cc89df3d757d10dec777061452 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= Date: Wed, 24 Nov 2021 10:50:19 +0100 Subject: [PATCH] Fix nodes_leaves for base_estimator --- odte/Odte.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/odte/Odte.py b/odte/Odte.py index 890f649..e1526b9 100644 --- a/odte/Odte.py +++ b/odte/Odte.py @@ -88,13 +88,14 @@ class Odte(BaseEnsemble, ClassifierMixin): return self def _compute_metrics(self) -> None: - tdepth = tnodes = tleaves = 0 + tdepth = tnodes = tleaves = 0.0 for estimator in self.estimators_: - nodes, leaves = estimator.nodes_leaves() - depth = estimator.depth_ - tdepth += depth - tnodes += nodes - tleaves += leaves + if hasattr(estimator, "nodes_leaves"): + nodes, leaves = estimator.nodes_leaves() + depth = estimator.depth_ + tdepth += depth + tnodes += nodes + tleaves += leaves self.depth_ = tdepth / self.n_estimators self.leaves_ = tleaves / self.n_estimators self.nodes_ = tnodes / self.n_estimators