Doctorado-ML/benchmark
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/benchmark.git synced 2025-08-16 16:05:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Makefile

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2022-01-15 01:15:10 +01:00
parent f43622504c
commit 2c901d761c
2 changed files with 33 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/Models.py
View File

@@ -43,9 +43,11 @@ class Models:
nodes = 0
leaves = result.get_n_leaves()
depth = 0
elif name=="Bagging":
elif name == "Bagging":
if hasattr(result.base_estimator_, "nodes_leaves"):
nodes, leaves = list(zip(*[x.nodes_leaves() for x in result.estimators_]))
nodes, leaves = list(
zip(*[x.nodes_leaves() for x in result.estimators_])
)
nodes, leaves = mean(nodes), mean(leaves)
depth = mean([x.depth_ for x in result.estimators_])
elif hasattr(result.base_estimator_, "tree_"):
@@ -53,7 +55,7 @@ class Models:
leaves = mean([x.get_n_leaves() for x in result.estimators_])
depth = mean([x.get_depth() for x in result.estimators_])
else:
nodes = leaves=depth=0
nodes = leaves = depth = 0
elif name == "RandomForest":
leaves = mean([x.get_n_leaves() for x in result.estimators_])
depth = mean([x.get_depth() for x in result.estimators_])