mirror of
https://github.com/Doctorado-ML/benchmark.git
synced 2025-08-18 00:45:54 +00:00
Add score hyperparameter
This commit is contained in:
10
src/main.py
10
src/main.py
@@ -8,6 +8,13 @@ from Results import Report
|
||||
|
||||
def parse_arguments():
|
||||
ap = argparse.ArgumentParser()
|
||||
ap.add_argument(
|
||||
"-s",
|
||||
"--score",
|
||||
type=str,
|
||||
required=True,
|
||||
help="score name {accuracy, f1_macro, ...}",
|
||||
)
|
||||
ap.add_argument(
|
||||
"-P",
|
||||
"--platform",
|
||||
@@ -55,6 +62,7 @@ def parse_arguments():
|
||||
)
|
||||
args = ap.parse_args()
|
||||
return (
|
||||
args.score,
|
||||
args.model,
|
||||
args.n_folds,
|
||||
args.platform,
|
||||
@@ -66,6 +74,7 @@ def parse_arguments():
|
||||
|
||||
|
||||
(
|
||||
score,
|
||||
model,
|
||||
folds,
|
||||
platform,
|
||||
@@ -75,6 +84,7 @@ def parse_arguments():
|
||||
report,
|
||||
) = parse_arguments()
|
||||
job = Experiment(
|
||||
score_name=score,
|
||||
model_name=model,
|
||||
datasets=Datasets(),
|
||||
hyperparams_dict=hyperparameters,
|
||||
|
Reference in New Issue
Block a user