添加: 多标签文本分类
将自己的数据处理成两个文件:数据文件,标签文件。
数据文件存放于data/corpus/{dataset}.txt
,标签文件存放于data/{dataset}.txt
, 格式是id
, train/text标识
,(多)类别
python build_graph {dataset}
创建图特征数据python train.py {dataset} True
进行多标签任务训练
原始repo:
原始论文:
Liang Yao, Chengsheng Mao, Yuan Luo. "Graph Convolutional Networks for Text Classification." In 33rd AAAI Conference on Artificial Intelligence (AAAI-19), 7370-7377
Python 3.6
Tensorflow >= 1.4.0