/LR-GCN

Label-Representative Graph Convolutional Network for Multi-Label Text Classification

Primary LanguagePython

Label-Representative Graph Convolutional Network for Multi-Label Text Classification

This repository is the official implementation of our LR-GCN (Label-Representative Graph Convolutional Network for Multi-Label Text Classification) paper.

Requirements

To install requirements:

pip install -r requirements.txt

Usage tutorial

To train the model

cd trainer
python main.py