File tree Expand file tree Collapse file tree 1 file changed +14
-1
lines changed
Expand file tree Collapse file tree 1 file changed +14
-1
lines changed Original file line number Diff line number Diff line change 2626from model_scoring .run_scoring import batch_process_models
2727from model_scoring .utils .logging import configure_console_only_logging
2828from model_scoring .utils .config_loader import load_config_from_path
29- from model_scoring .csv_reporter import generate_csv_report
29+ from model_scoring .utils .csv_reporter import generate_csv_report
30+ from model_scoring .utils .graph_reporter import generate_report as generate_graph_report
3031from config import scoring_config as default_scoring_config
3132
3233# ------------------------------------------------------------------------------------------------
@@ -71,6 +72,12 @@ def parse_args() -> argparse.Namespace:
7172 help = 'Generate a CSV report from the results.'
7273 )
7374
75+ parser .add_argument (
76+ '--graph' ,
77+ action = 'store_true' ,
78+ help = 'Generate an HTML graph report from the results.'
79+ )
80+
7481 parser .add_argument (
7582 "--version" ,
7683 action = "version" ,
@@ -97,6 +104,12 @@ def main() -> None:
97104 print ("[*] CSV report generated successfully." )
98105 return
99106
107+ if args .graph :
108+ print ("[*] Generating HTML graph report..." )
109+ generate_graph_report ()
110+ print ("[*] HTML graph report generated successfully." )
111+ return
112+
100113 # Load scoring configuration
101114 if args .config :
102115 print (f"[*] Loading custom configuration from: { args .config } " )
You can’t perform that action at this time.
0 commit comments