Fix compute number of nodes

This commit is contained in:
2021-04-13 22:31:05 +02:00
parent 783d105099
commit b55f59a3ec
2 changed files with 4 additions and 5 deletions

View File

@@ -841,10 +841,9 @@ class Stree(BaseEstimator, ClassifierMixin):
nodes = 0
leaves = 0
for node in self:
nodes += 1
if node.is_leaf():
leaves += 1
else:
nodes += 1
return nodes, leaves
def __iter__(self) -> Siterator: