To install the necessary dependencies, run:
pip install -r requirements.txt
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.
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