Installation

To install the necessary dependencies, run:

pip install -r requirements.txt

Running the experiments

To run the experiments, use the following command:

python main.py -cn active dataset=DATASET heuristic=AL_METHOD

The list of available datasets and active learning methods can be found in the conf/dataset and conf/heuristic directories, respectively. For other configuration options, please refer to the conf/active.yaml file. Our method DIAL is named QBC in the configuration files.

Acknowledgements

Most of the code is based on the Baal and SubpopBench. The implementation of BADGE and BAIT is based on the repo badge and Cluster-Margin is based on the repo deep-active-learning