Skip to content

Commit d10d966

Browse files
authored
Update score_models.py to v0.6.0
1 parent 18844ff commit d10d966

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

score_models.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,8 @@
2626
from model_scoring.run_scoring import batch_process_models
2727
from model_scoring.utils.logging import configure_console_only_logging
2828
from 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
3031
from 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}")

0 commit comments

Comments
 (0)