chiennv2000 / GCN-MTC

This repository is the official implementation of our paper.

Label Correlation Based Graph Convolutional Network for Multi-label Text Classification

This repository is the official implementation of our GCN-MTC paper.

Requirements

To install requirements:

pip install -r requirements.txt

Training

To train the model(s) in the paper on RCV1, run this command:

python train.py

Evaluation

To test my model on RCV1, run:

python eval.py --checkpoint <path-to-checkpoint-file> --graph_feature <path-to-feature-of-graph>