Skip to content

Home

Logo dark mode Logo light mode

Baal is a Bayesian active learning library. We provide methods to estimate sampling from the posterior distribution in order to maximize the efficiency of labelling during active learning. Our library is suitable for research and industrial applications.

To know more on what is Bayesian active learning, see our User guide.

We are a member of Pytorch's ecosystem, and we welcome contributions from the community. If you have any question, we are reachable on Slack.

Installation

Baal is available as a package on PyPI:

pip install baal

Additional dependencies for vision and NLP

baal[nlp] installs needed dependencies for HuggingFace support.

baal[vision] installs dependencies for our Lightning-Flash integration.

Support

For support, we have several ways to help you: