From ec57d112b92ab0b18704b8112ad2a7838b3bc0af Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= Date: Wed, 30 Jun 2021 14:53:03 +0200 Subject: [PATCH] Update get depth of STree in report_score --- report_score.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/report_score.py b/report_score.py index a3e0e05..079d765 100644 --- 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)