Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Add the support of testing on multiple datasets after training on one dataset. #69

Merged
merged 5 commits into from
Dec 19, 2023

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
3 checks passed
@Harry-zzh Harry-zzh deleted the sem_seg branch December 20, 2023 02:33
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants