Skip to content

Commit f2f84ad

Browse files
Report config selector (#107)
* report configs in runner * space in help * default and const value * pep8 * retrigger checks * old name * update help
1 parent a4895a7 commit f2f84ad

File tree

2 files changed

+38
-3
lines changed

2 files changed

+38
-3
lines changed
Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
{
2+
"header": [
3+
"algorithm",
4+
"stage",
5+
"device",
6+
"input_data:data_order",
7+
"input_data:data_type",
8+
"input_data:dataset_name",
9+
"input_data:classes",
10+
"algorithm_parameters:tol",
11+
"algorithm_parameters:max_iter",
12+
"algorithm_parameters:solver",
13+
"algorithm_parameters:C",
14+
"algorithm_parameters:kernel",
15+
"algorithm_parameters:nu",
16+
"algorithm_parameters:eps",
17+
"algorithm_parameters:n_neighbors",
18+
"algorithm_parameters:n_estimators",
19+
"algorithm_parameters:n_clusters",
20+
"algorithm_parameters:min_samples",
21+
"algorithm_parameters:fit_intercept",
22+
"algorithm_parameters:max_depth",
23+
"algorithm_parameters:max_features"
24+
],
25+
"comparison_method": {
26+
"default": "2 / 1"
27+
},
28+
"aggregation_metrics": [
29+
"geomean"
30+
]
31+
}

runner.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,9 +71,13 @@ def get_configs(path: Path) -> List[str]:
7171
parser.add_argument('--verbose', default='INFO', type=str,
7272
choices=("ERROR", "WARNING", "INFO", "DEBUG"),
7373
help='Print additional information during benchmarks running')
74-
parser.add_argument('--report', default=False, action='store_true',
74+
parser.add_argument('--report', nargs='?', default=None, metavar='ConfigPath', type=str,
75+
const='report_generator/default_report_gen_config.json',
7576
help='Create an Excel report based on benchmarks results. '
76-
'Need "openpyxl" library')
77+
'If the parameter is not set, the reporter will not be launched. '
78+
'If the parameter is set and the config is not specified, '
79+
'the default config will be used. '
80+
'Need "openpyxl" library')
7781
args = parser.parse_args()
7882

7983
logging.basicConfig(
@@ -278,7 +282,7 @@ class GenerationArgs:
278282
command = 'python report_generator/report_generator.py ' \
279283
+ f'--result-files {name_result_file} ' \
280284
+ f'--report-file {name_result_file}.xlsx ' \
281-
+ '--generation-config report_generator/default_report_gen_config.json'
285+
+ '--generation-config ' + args.report
282286
logging.info(command)
283287
stdout, stderr = utils.read_output_from_command(command)
284288
if stderr != '':

0 commit comments

Comments
 (0)