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

Update get depth of STree in report_score

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2021-06-30 14:53:03 +02:00
parent b65ccdda1d
commit ec57d112b9
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
report_score.py
View File

@@ -145,7 +145,9 @@ def process_dataset(dataset, verbose, model, params):
leaves_item = result_item.get_n_leaves() leaves_item = result_item.get_n_leaves()
else: else:
nodes_item, leaves_item = result_item.nodes_leaves() nodes_item, leaves_item = result_item.nodes_leaves()
depth_item = result_item.depth_ depth_item = (
result_item.depth_ if hasattr(result_item, "depth_") else 0
)
nodes.append(nodes_item) nodes.append(nodes_item)
leaves.append(leaves_item) leaves.append(leaves_item)
depths.append(depth_item) depths.append(depth_item)