Skip to content
New issue

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

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

Already on GitHub? Sign in to your account

Support per-metric model specification in MBM #3009

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

sdaulton
Copy link
Contributor

@sdaulton sdaulton commented Nov 1, 2024

Summary:
Enables using different models for different metrics.

  • Adds ModelConfig dataclass to specify a single botorch model configuration
  • Adds a list of ModelConfigs to SurrogateSpec
  • Adds a dictionary mapping metric names to list of ModelConfigs to enable per-metric model specification
  • Lists of model configs are used to enable per-metric model selection across multiple ModelConfigs in a subsequent diff.

Reviewed By: saitcakmak

Differential Revision: D64793595

Summary:
Enables using different models for different metrics.

* Adds ModelConfig dataclass to specify a single botorch model configuration
* Adds a list of ModelConfigs to SurrogateSpec
* Adds a dictionary mapping metric names to list of ModelConfigs to enable per-metric model specification
* Lists of model configs are used to enable per-metric model selection across multiple ModelConfigs in a subsequent diff.

Reviewed By: saitcakmak

Differential Revision: D64793595
@facebook-github-bot facebook-github-bot added the CLA Signed Do not delete this pull request or issue due to inactivity. label Nov 1, 2024
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D64793595

@codecov-commenter
Copy link

Codecov Report

Attention: Patch coverage is 97.56098% with 3 lines in your changes missing coverage. Please review.

Project coverage is 95.60%. Comparing base (fcfda95) to head (7fb4c59).

Files with missing lines Patch % Lines
ax/models/torch/botorch_modular/model.py 71.42% 2 Missing ⚠️
ax/models/torch/botorch_modular/surrogate.py 96.96% 1 Missing ⚠️
Additional details and impacted files
@@           Coverage Diff           @@
##             main    #3009   +/-   ##
=======================================
  Coverage   95.60%   95.60%           
=======================================
  Files         483      483           
  Lines       48907    48989   +82     
=======================================
+ Hits        46758    46837   +79     
- Misses       2149     2152    +3     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed Do not delete this pull request or issue due to inactivity. fb-exported
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants