Skip to content

Implemented BALDAcquisition with testcases#15

Open
aditiagarwal-02 wants to merge 1 commit intosustainability-lab:mainfrom
aditiagarwal-02:baldacquisition
Open

Implemented BALDAcquisition with testcases#15
aditiagarwal-02 wants to merge 1 commit intosustainability-lab:mainfrom
aditiagarwal-02:baldacquisition

Conversation

@aditiagarwal-02
Copy link

Implementing BALD Acquisition Function

Description:
This PR introduces the implementation of the BALD (Bayesian Active Learning by Disagreement) acquisition function proposed by Gal et. al.(2017)

Implementation Details:

Added the BALDAcquisition class in astra/torch/al/acquisitions/furthest.py
Modified astra/torch/al/strategies/mc.py (minor bug fixes)

Testing:

Testcase added in tests/torch/acquisitions/test_bald.py

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.

1 participant