partial_label_masking

TensorFlow implementation of Partial Label Masking (PLM)

Prerequisite

• Poetry >= 1.1.12
• GNU Make

Installation

make dep

Example

See examples/mlrs_net/train_model_plm.py

Format, lint and test

make format
make lint
make test

Reference

K. Duarte, Y. Rawat and M. Shah, "PLM: Partial Label Masking for Imbalanced Multi-Label Classification," CVPR2021 Workshops, pp. 2739-2748.