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

Ax sweeper generation strategy feature #2978

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

lwelzel
Copy link

@lwelzel lwelzel commented Oct 29, 2024

Add the possibility to configure the generation strategy of the sweeper in the ax-sweeper plugin.

Motivation

I want to be able to define a generation strategy with specific steps in the config for the ax-sweeper. See issue #2977

Have you read the Contributing Guidelines

Yes

Test Plan

The tests could be similar to the current tests for the ax-sweeper. I am not sure how this works with the old versions of the ax-platform and botorch since using their more recent versions lead to some errors because the ax-sweeper config ExperimentConfig dataclass was misconfigured (ExperimentConfig.minimize and ExperimentConfig.objective_name).

Related Issues and PRs

Feature request: #2977

lwelzel and others added 2 commits October 29, 2024 14:19
Implements configs for both ax.modelbridge.generation_strategy.GenerationStrategy and GenerationStep in the ax-sweeper config.py and a constructor method in the CoreAxSweeper class.
@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Oct 29, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants