Skip to content

Repository for the paper IEEE Access: Dealing with Multi-Positive Unlabelled learning combining metric learning and deep clustering

Notifications You must be signed in to change notification settings

ml-unito/Multi-class-unlabeled-learning

Repository files navigation

Prototype based Multi-Positive and Unlabelled Learning approach

Python implementation for the paper:

Usage

  1. Install the required packages.
    pip install -r requirements.txt
  2. Clone the code.
  3. Prepare datasets. All datasets are already given with the code except for reuters. For this one you have to download the files running the code written in 'data/reuters/get_data.sh'
  4. Run the code!
    python main_experiments.py if you want to compare the competing methods on a standard setting
    python ablation_labeled_samples.py if you want to compare the competing methods with a varying number of labeled samples
    python ablation_positive_classes.py if you want to compare the competing methods with a varying number of positive classes
    python ablation_protoMPUL.py if you want to study the performances of each stage of ProtoMPUL (ablation study)

You can customize experiments passing arguments to the interpreter. For example, if you want reproducible results, you have to specify
python main_experiments.py --generate_dataset=False
in this way the folds generated for each experiment will remain always the same. Inspect the code if you want to get more information

Code is free of use!

About

Repository for the paper IEEE Access: Dealing with Multi-Positive Unlabelled learning combining metric learning and deep clustering

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published