Skip to content
forked from YUHANYU/LGSM

This is the code for the Paper: Label Sequence Generation Model for Multi-Label Learning

Notifications You must be signed in to change notification settings

fightingforcode/LGSM

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

LGSM

This is the code of LSGM. The default multi-label learning dataset is Yeast. If you want to run this program, please use the command "python main.py".
Please note that if you use other data sets, you may need to modify the learning rate to get good results.

Note

The code is not highly optimized, so the runtime will be very long. It takes about 6 hours to get the final result. Please wait patiently.

How to use

Excute the command: python main.py If you want to modify the mll dataset or other parameters, please modify the main.py or parameters.py

Enviroment

pytorch 0.4.1 CUDA 9.0 scikit learn 0.19

About

This is the code for the Paper: Label Sequence Generation Model for Multi-Label Learning

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%