add --export-raw-predictions

2 files changed, 12 insertions, 2 deletions

diff --git a/bin/analyze-archive.py b/bin/analyze-archive.py
index a725833..4aab326 100755
--- a/bin/analyze-archive.py
+++ b/bin/analyze-archive.py
@@ -910,7 +910,12 @@ if __name__ == "__main__":
             lambda m: m.get_fitted()[0], xv_count
         )
     else:
-        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=NpEncoder)
+        else:
+            analytic_quality = model.assess(param_model)
         xv_analytic_models = [model]
 
     if "tex" in show_models or "tex" in show_quality:
diff --git a/bin/analyze-kconfig.py b/bin/analyze-kconfig.py
index 1d03839..aab71e8 100755
--- a/bin/analyze-kconfig.py
+++ b/bin/analyze-kconfig.py
@@ -258,7 +258,12 @@ def main():
         )
     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)
         xv_analytic_models = [model]
         if lut_model:
             lut_quality = model.assess(lut_model)