autogluon / tabrepo

Apache License 2.0
36 stars 9 forks source link

Add TaskMetadata and EnsembleScorer #32

Closed Innixma closed 12 months ago

Innixma commented 1 year ago

Add TaskMetadata and EnsembleScorer to cleanup and streamline the logic in ensemble_selection_config_scorer.py.

The overall logic remains the same, but with a more clean structure. I've confirmed that simulation results remain identical.

Note: GitHub makes this look like a much bigger change than it is. What changed is that the logic in run_task was moved to a dedicated class called EnsembleScorer, which is constructed during init of EnsembleSelectionConfigScorer. TaskMetadata is a convenience wrapper around zeroshot_gt to simplify the code.