diff options
Diffstat (limited to 'bin/analyze-archive.py')
| -rwxr-xr-x | bin/analyze-archive.py | 7 |
1 files changed, 6 insertions, 1 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: |