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.

Baal 2.0 !

Baal is now version 2.0! We made a lot of changes to make everyone life easier! See our Release Note for details.

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.