Skip to content

Conversation

Harry-zzh
Copy link
Contributor

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@Harry-zzh Harry-zzh changed the title Modify exec.py Add the support of testing on multiple datasets after training on one dataset. Dec 11, 2023
@@ -193,6 +193,7 @@ def run(
"problem_type": train_data.problem_type,
"presets": params.pop("presets", None),
"path": os.path.join(benchmark_dir, "models"),
"eval_metric": val_data.metric,
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

val_data might not be available for all datasets so we need to handle that.

Copy link
Contributor Author

@Harry-zzh Harry-zzh Dec 15, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

How about adding these lines:

if val_data is not None:
    predictor_args["eval_metric"] = val_data.metric

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes that would work.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done.

@suzhoum suzhoum merged commit ea85c04 into autogluon:master Dec 19, 2023
@Harry-zzh Harry-zzh deleted the sem_seg branch December 20, 2023 02:33
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants