mirror of
https://github.com/Doctorado-ML/benchmark.git
synced 2025-08-18 00:45:54 +00:00
Fix be_main args issue
This commit is contained in:
@@ -206,7 +206,16 @@ class Arguments:
|
|||||||
"help": "Use best hyperparams file?",
|
"help": "Use best hyperparams file?",
|
||||||
},
|
},
|
||||||
],
|
],
|
||||||
"platform": [],
|
"platform": [
|
||||||
|
("-P", "--platform"),
|
||||||
|
{
|
||||||
|
"action": EnvDefault,
|
||||||
|
"envvar": "platform",
|
||||||
|
"type": str,
|
||||||
|
"required": True,
|
||||||
|
"help": "Platform where the test is run",
|
||||||
|
},
|
||||||
|
],
|
||||||
"quiet": [
|
"quiet": [
|
||||||
("-q", "--quiet"),
|
("-q", "--quiet"),
|
||||||
{
|
{
|
||||||
|
@@ -2,7 +2,7 @@
|
|||||||
import os
|
import os
|
||||||
from benchmark.Experiments import Experiment, Datasets
|
from benchmark.Experiments import Experiment, Datasets
|
||||||
from benchmark.Results import Report
|
from benchmark.Results import Report
|
||||||
from Arguments import Arguments
|
from benchmark.Arguments import Arguments
|
||||||
|
|
||||||
"""Do experiment and build result file, optionally print report with results
|
"""Do experiment and build result file, optionally print report with results
|
||||||
"""
|
"""
|
||||||
@@ -28,15 +28,14 @@ def main():
|
|||||||
grid_paramfile=args.grid_paramfile,
|
grid_paramfile=args.grid_paramfile,
|
||||||
progress_bar=not args.quiet,
|
progress_bar=not args.quiet,
|
||||||
platform=args.platform,
|
platform=args.platform,
|
||||||
title=args.experiment_title,
|
title=args.title,
|
||||||
folds=args.folds,
|
folds=args.n_folds,
|
||||||
)
|
)
|
||||||
job.do_experiment()
|
job.do_experiment()
|
||||||
if report:
|
if report:
|
||||||
result_file = job.get_output_file()
|
result_file = job.get_output_file()
|
||||||
report = Report(result_file)
|
report = Report(result_file)
|
||||||
report.report()
|
report.report()
|
||||||
|
|
||||||
if args.dataset is not None:
|
if args.dataset is not None:
|
||||||
print(f"Partial result file removed: {result_file}")
|
print(f"Partial result file removed: {result_file}")
|
||||||
os.remove(result_file)
|
os.remove(result_file)
|
||||||
|
Reference in New Issue
Block a user