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)