cMAB Fast Update via Variational Inference #48

shaharbar1 · 2024-09-02T10:09:31Z

Changes

Edited BaseBayesianLogisticRegression and inheritors on model.py to support variational inference by adding fast_inference control parameter on class attributes and adding control arguments on update method.
Edited update UTs on test_cmab.py to support new inference mode.
Edited cMABs cold start function tto support new inference mode.
Removed redundant test_execution_time.py.
Edited version on pyproject.toml.

pybandits/cmab.py

pybandits/model.py

pybandits/cmab.py

pybandits/model.py

### Changes * Edited BaseBayesianLogisticRegression and inheritors on model.py to support variational inference by adding fast_inference control parameter on class attributes and adding control arguments on update method. * Edited BaseBayesianLogisticRegression to allow faster update via vectorization of PyMC operations. * Edited "update" UTs on test_cmab.py to support new inference mode. * Edited cMABs cold start function tto support new inference mode. * Removed redundant test_execution_time.py. * Edited version on pyproject.toml.

dariodandrea

Great job @shaharbar1 !

shaharbar1 requested review from adarmiento, dariodandrea and j3rom3c September 2, 2024 10:09

shaharbar1 force-pushed the feature/variational_infernce_cmab branch from 52719f8 to a7ad3ac Compare September 3, 2024 12:24

adarmiento reviewed Sep 3, 2024

View reviewed changes

pybandits/cmab.py Outdated Show resolved Hide resolved

pybandits/cmab.py Outdated Show resolved Hide resolved

pybandits/model.py Outdated Show resolved Hide resolved

shaharbar1 force-pushed the feature/variational_infernce_cmab branch 7 times, most recently from e2e11ee to f6ffd58 Compare September 4, 2024 11:52

shaharbar1 added the enhancement New feature or request label Sep 4, 2024

shaharbar1 force-pushed the feature/variational_infernce_cmab branch 2 times, most recently from bdf207e to 5fc5a94 Compare September 23, 2024 08:50