ESL-NN is a "Hierarchical Multi-Label Classification System" based on PyTorch which is free for download and free for non-commecial use.
- Multi-class classification
- Multi-label classification
- Hierarchical multi-label classification with tree and DAG structure.
- Linux
- MACOS
- Peerapon Vateekul (Supervisor)
- Chanatip Saetia
/assemble_classifier
- Main classifiers code which store ESLNN, SHLNN, HMCLMLP/classifier
- Classifiers of each level code which will be composed together in main classifiers in/assemble_classifier
/data
- Data preprocessing code and Dataset class which is specifically used for our main classifier/embedding
- Document embedding code/evaluation
- Evaluation metric codeTrain.ipynb
- A demo jupyter notebook for a training processTest.ipynb
- A demo jupyter notebook for a testing process