Skip to content
forked from adrinta/MAGNET

MAGNET: Multi-Label Text Classification using Attention-based Graph Neural Network

Notifications You must be signed in to change notification settings

sherylxun/MAGNET

 
 

Repository files navigation

MAGNET: Multi-Label Text Classification using Attention-based Graph Neural Network

Under development to implement this paper https://arxiv.org/abs/2003.11644 Using PyTorch

This is the result of what i am doing so far:

Embedding Dataset Seq Length Train Val Split Micro-F1 Score (Validation)
GloVe Reuters-21578 128 7769 / 3019 0.8523±

About

MAGNET: Multi-Label Text Classification using Attention-based Graph Neural Network

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 93.5%
  • Python 6.5%