From 5fd55e1a09b5558064c6eac77b2c5b39624a0739 Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Fri, 22 Dec 2023 07:54:05 +0100 Subject: analyze-log: add --export-raw-predictions support --- bin/analyze-log.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'bin') diff --git a/bin/analyze-log.py b/bin/analyze-log.py index 21db337..4c37928 100755 --- a/bin/analyze-log.py +++ b/bin/analyze-log.py @@ -220,7 +220,12 @@ def main(): analytic_quality, _ = xv.kfold(lambda m: m.get_fitted()[0], xv_count) else: static_quality = model.assess(static_model) - analytic_quality = model.assess(param_model) + if args.export_raw_predictions: + analytic_quality, raw_results = model.assess(param_model, return_raw=True) + with open(args.export_raw_predictions, "w") as f: + json.dump(raw_results, f, cls=dfatool.utils.NpEncoder) + else: + analytic_quality = model.assess(param_model) if "static" in args.show_model or "all" in args.show_model: print("--- static model ---") -- cgit v1.2.3