/Seq2Set

Code for the paper "A Deep Reinforced Sequence-to-Set Model for Multi-Label Classification"

Primary LanguagePython

Seq2Set

Code for the paper "A Deep Reinforced Sequence-to-Set Model for Multi-Label Classification". Please use the MLE method to pre-train the model and use -restore to load the pre-trained checkpoint.