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

Branin and Hartmann proxies #297

Merged
merged 21 commits into from
Mar 14, 2024
Merged

Conversation

alexhernandezgarcia
Copy link
Owner

@alexhernandezgarcia alexhernandezgarcia commented Mar 4, 2024

This PR implements the Branin and Hartmann proxies, functions widely used in the optimisation and (multi-fidelity) active learning literature, including the Multi-Fidelity Active Learning with GFlowNets paper.

The Branin function is defined on the real plane and the Hartmann function on a 6D real hyper-plane. Therefore, these functions (proxies) can be used to train both grid (discrete) and cube (continuous) environments.

They rely on the botorch implementation. Therefore botorch has been added as a new dependency.

Both proxies can be turned into multi-fidelity oracles via the parameter fidelity.

This PR includes a bit of additional clean-up and refactoring:

  • Remove the deprecated variable func from environment config files.
  • Remove the variable higher_is_better from proxy config files.
  • Remove variable proxy_factor in the base environment, since it is assumed that proxy outputs will be negative and lower is better. If not, this can still be controlled with the beta parameter. In any case, this aspect should be revised in the future.
  • Move all grid and cube proxies into /gflownet/proxy/box and similarly for the configuration files.
  • Extends docstring.

@josephdviviano
Copy link
Collaborator

josephdviviano commented Mar 13, 2024 via email

@alexhernandezgarcia alexhernandezgarcia changed the title [WIP] Branin and Hartmann proxies Branin and Hartmann proxies Mar 13, 2024
@alexhernandezgarcia alexhernandezgarcia changed the base branch from main to activelearning March 14, 2024 14:53
@alexhernandezgarcia alexhernandezgarcia merged commit 4fa1bb3 into activelearning Mar 14, 2024
1 check passed
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