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.

Sign up for GitHub

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

Add batched acquisition functions and knowledge gradient #37

Merged
merged 37 commits into from
Sep 2, 2023
Merged

Add batched acquisition functions and knowledge gradient #37

merged 37 commits into from
Sep 2, 2023

Commits on Aug 13, 2023

  1. Add utility to split dict in batches

    @ziatdinovmax
    ziatdinovmax committed Aug 13, 2023
    Configuration menu
    Copy the full SHA
    13ccae5 View commit details
    Browse the repository at this point in the history

Commits on Aug 15, 2023

  1. Streamline vmap compute in vi DKL

    @ziatdinovmax
    ziatdinovmax committed Aug 15, 2023
    Configuration menu
    Copy the full SHA
    39c392e View commit details
    Browse the repository at this point in the history

  2. fix batch prediction in viDKL 

    this is in response to previous commit
    @ziatdinovmax
    ziatdinovmax committed Aug 15, 2023
    Configuration menu
    Copy the full SHA
    742ea51 View commit details
    Browse the repository at this point in the history

  3. Update tests

    @ziatdinovmax
    ziatdinovmax committed Aug 15, 2023
    Configuration menu
    Copy the full SHA
    06fa254 View commit details
    Browse the repository at this point in the history

  4. Update the args sequence in get_mvn_posterior

    @ziatdinovmax
    ziatdinovmax committed Aug 15, 2023
    Configuration menu
    Copy the full SHA
    2cea61e View commit details
    Browse the repository at this point in the history

  5. fix sample_from_posterior 

    in response to earlier change in the arg sequence
    @ziatdinovmax
    ziatdinovmax committed Aug 15, 2023
    Configuration menu
    Copy the full SHA
    6516994 View commit details
    Browse the repository at this point in the history

Commits on Aug 16, 2023

  1. Add 'get_samples' for viDKL

    @ziatdinovmax
    ziatdinovmax committed Aug 16, 2023
    Configuration menu
    Copy the full SHA
    484f70c View commit details
    Browse the repository at this point in the history

  2. Add utility for random sampling dict with params

    @ziatdinovmax
    ziatdinovmax committed Aug 16, 2023
    Configuration menu
    Copy the full SHA
    2cff802 View commit details
    Browse the repository at this point in the history

Commits on Aug 20, 2023

  1. Refactor acquisition functions (work in progress)

    @ziatdinovmax
    ziatdinovmax committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    7ba47c3 View commit details
    Browse the repository at this point in the history

  2. fix tests

    @ziatdinovmax
    ziatdinovmax committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    ed68fe3 View commit details
    Browse the repository at this point in the history

  3. Refactor acquisition functions 

    - Add batche-mode acquisition functions
- Move ei, ucb, and poi to base_acq
- Update tests
    @ziatdinovmax
    ziatdinovmax committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    9720a86 View commit details
    Browse the repository at this point in the history

  4. Remove redundant rng_key

    @ziatdinovmax
    ziatdinovmax committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    df1b80f View commit details
    Browse the repository at this point in the history

  5. check input dimensionality

    @ziatdinovmax
    ziatdinovmax committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    cf52a6d View commit details
    Browse the repository at this point in the history

  6. Update tests

    @ziatdinovmax
    ziatdinovmax committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    8fe9e24 View commit details
    Browse the repository at this point in the history

  7. remove compute_batch_acqusition duplicate

    @ziatdinovmax
    ziatdinovmax committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    60149d9 View commit details
    Browse the repository at this point in the history

  8. (re)-add pure uncertainty-based explroation

    @ziatdinovmax
    ziatdinovmax committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    f732710 View commit details
    Browse the repository at this point in the history

  9. Fix bug in UE 

    After the last update it was actually computing ucb
    @ziatdinovmax
    ziatdinovmax committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    506c5ba View commit details
    Browse the repository at this point in the history

Commits on Aug 24, 2023

  1. Update self.get_mvn_posterior 

    - remove jit decorator (prevents KG implementatio)
- Do not pop noise out of the dictionary
    @ziatdinovmax
    ziatdinovmax committed Aug 24, 2023
    Configuration menu
    Copy the full SHA
    b4fc9f0 View commit details
    Browse the repository at this point in the history

  2. fix typo

    @ziatdinovmax
    ziatdinovmax committed Aug 24, 2023
    Configuration menu
    Copy the full SHA
    218c0a7 View commit details
    Browse the repository at this point in the history

  3. Exclude "noise" from vmap in multi-task kernel 

    (it might make sense to do it for other kernels as well)
    @ziatdinovmax
    ziatdinovmax committed Aug 24, 2023
    Configuration menu
    Copy the full SHA
    e5e37d7 View commit details
    Browse the repository at this point in the history

  4. Add base KG implementation

    @ziatdinovmax
    ziatdinovmax committed Aug 24, 2023
    Configuration menu
    Copy the full SHA
    c07c636 View commit details
    Browse the repository at this point in the history

Commits on Aug 26, 2023

  1. Knowledge gradient updates

    @ziatdinovmax
    ziatdinovmax committed Aug 26, 2023
    Configuration menu
    Copy the full SHA
    00a18a2 View commit details
    Browse the repository at this point in the history

  2. Update args and docstrings in KG

    @ziatdinovmax
    ziatdinovmax committed Aug 26, 2023
    Configuration menu
    Copy the full SHA
    bfcf305 View commit details
    Browse the repository at this point in the history

Commits on Aug 27, 2023

  1. Fix KG for minimization problems

    @ziatdinovmax
    ziatdinovmax committed Aug 27, 2023
    Configuration menu
    Copy the full SHA
    15b19b6 View commit details
    Browse the repository at this point in the history

  2. Add batch-mode KG (qKG)

    @ziatdinovmax
    ziatdinovmax committed Aug 27, 2023
    Configuration menu
    Copy the full SHA
    065a8e9 View commit details
    Browse the repository at this point in the history

  3. Update docstrings

    @ziatdinovmax
    ziatdinovmax committed Aug 27, 2023
    Configuration menu
    Copy the full SHA
    235a515 View commit details
    Browse the repository at this point in the history

  4. update docstrings

    @ziatdinovmax
    ziatdinovmax committed Aug 27, 2023
    Configuration menu
    Copy the full SHA
    315cb1c View commit details
    Browse the repository at this point in the history

  5. Update docstrings

    @ziatdinovmax
    ziatdinovmax committed Aug 27, 2023
    Configuration menu
    Copy the full SHA
    af43280 View commit details
    Browse the repository at this point in the history

  6. Update docstrings

    @ziatdinovmax
    ziatdinovmax committed Aug 27, 2023
    Configuration menu
    Copy the full SHA
    3f75e71 View commit details
    Browse the repository at this point in the history

Commits on Aug 31, 2023

  1. Use jax.numpy for splitting dict with hmc params

    @ziatdinovmax
    ziatdinovmax committed Aug 31, 2023
    Configuration menu
    Copy the full SHA
    23c08c9 View commit details
    Browse the repository at this point in the history

  2. Update random_sampled_dict

    @ziatdinovmax
    ziatdinovmax committed Aug 31, 2023
    Configuration menu
    Copy the full SHA
    a74389d View commit details
    Browse the repository at this point in the history

  3. Refactor acquisition functions

    @ziatdinovmax
    ziatdinovmax committed Aug 31, 2023
    Configuration menu
    Copy the full SHA
    5f37fe6 View commit details
    Browse the repository at this point in the history

  4. Update tests

    @ziatdinovmax
    ziatdinovmax committed Aug 31, 2023
    Configuration menu
    Copy the full SHA
    bc10e8f View commit details
    Browse the repository at this point in the history

  5. Update tests

    @ziatdinovmax
    ziatdinovmax committed Aug 31, 2023
    Configuration menu
    Copy the full SHA
    fe9a38a View commit details
    Browse the repository at this point in the history

Commits on Sep 1, 2023

  1. streamline maximiz_distance with vmap

    @ziatdinovmax
    ziatdinovmax committed Sep 1, 2023
    Configuration menu
    Copy the full SHA
    ed49d4e View commit details
    Browse the repository at this point in the history

Commits on Sep 2, 2023

  1. Update docstrings and imports

    @ziatdinovmax
    ziatdinovmax committed Sep 2, 2023
    Configuration menu
    Copy the full SHA
    005d167 View commit details
    Browse the repository at this point in the history

  2. Set default n_evals to 10

    @ziatdinovmax
    ziatdinovmax committed Sep 2, 2023
    Configuration menu
    Copy the full SHA
    5a8dad1 View commit details
    Browse the repository at this point in the history