Skip to content

A library for multi-class and multi-label classification

License

Notifications You must be signed in to change notification settings

ntumlgroup/LibMultiLabel

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

LibMultiLabel — a Library for Multi-class and Multi-label Classification

LibMultiLabel is a library for binary, multi-class, and multi-label classification. It has the following functionalities

  • end-to-end services from raw texts to final evaluation/analysis
  • support for common neural network architectures and linear classifiers
  • easy hyper-parameter selection

This is an on-going development so many improvements are still being made. Comments are very welcome.

Environments

  • Python: 3.8+
  • CUDA: 11.8, 12.1 (if training neural networks by GPU)
  • Pytorch: 2.0.1+

If you have a different version of CUDA, follow the installation instructions for PyTorch LTS at their website.

Documentation

See the documentation here: https://www.csie.ntu.edu.tw/~cjlin/libmultilabel