From 5adc39c8de1fbda6e1ba9abcaaaf17516eb046ae Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Wed, 10 Jan 2024 09:41:50 +0100 Subject: --show-model=param: add CART support --- bin/analyze-log.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'bin/analyze-log.py') diff --git a/bin/analyze-log.py b/bin/analyze-log.py index 8171cbb..b1dc6ba 100755 --- a/bin/analyze-log.py +++ b/bin/analyze-log.py @@ -229,6 +229,10 @@ def main(): info = param_info(name, attribute) if type(info) is df.AnalyticFunction: dfatool.cli.print_analyticinfo(f"{name:10s} {attribute:15s}", info) + elif type(info) is df.CARTFunction: + dfatool.cli.print_cartinfo( + f"{name:10s} {attribute:15s}", info, model.parameters + ) elif type(info) is df.SplitFunction: dfatool.cli.print_splitinfo( model.parameters, info, f"{name:10s} {attribute:15s}" -- cgit v1.2.3