Update get depth of STree in report_score

2025-08-16 07:56:07 +00:00 · 2021-06-30 14:53:03 +02:00
parent b65ccdda1d
commit ec57d112b9
1 changed files with 3 additions and 1 deletions
--- a/report_score.py
+++ b/report_score.py
@@ -145,7 +145,9 @@ def process_dataset(dataset, verbose, model, params):
                leaves_item = result_item.get_n_leaves()
            else:
                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)
            leaves.append(leaves_item)
            depths.append(depth_item)