From ddf6e00b3b16a07b994107a79450909f43588445 Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Tue, 20 Feb 2024 07:23:36 +0100 Subject: Re-add (very very basic, for now) Symbolic Regression support --- lib/cli.py | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'lib/cli.py') diff --git a/lib/cli.py b/lib/cli.py index 81fd9ae..499e959 100644 --- a/lib/cli.py +++ b/lib/cli.py @@ -124,6 +124,10 @@ def print_staticinfo(prefix, info): print(f"{prefix}: {info.value}") +def print_symreginfo(prefix, info): + print(f"{prefix}: {str(info.regressor)}") + + def print_cartinfo(prefix, info): _print_cartinfo(prefix, info.to_json()) @@ -203,6 +207,8 @@ def print_model(prefix, info): print_lmtinfo(prefix, info) elif type(info) is df.XGBoostFunction: print_xgbinfo(prefix, info) + elif type(info) is df.SymbolicRegressionFunction: + print_symreginfo(prefix, info) else: print(f"{prefix}: {type(info)} UNIMPLEMENTED") -- cgit v1.2.3