text_gcn

text_gcn

A implementação do texto GCN em nosso artigo:

Liang Yao, Chengsheng Mao, Yuan Luo. "Redes convolucionais de gráfico para classificação de texto". Na 33ª Conferência AAAI sobre Inteligência Artificial (AAAI-19), 7370-7377

Exigir

Python 2.7 ou 3.6

Tensorflow> = 1.4.0

Reproduzindo resultados

1. Execute python remove_words.py 20ng

2. Run python build_graph.py 20ng

3. Execute python train.py 20ng

4. Altere 20ng em linhas de comando acima de 3 para R8 , R52 , ohsumed e mr ao produzir resultados para outros conjuntos de dados.

Exemplo de dados de entrada

1. /data/20ng.txt indica nomes de documentos, divisão de treinamento/teste, etiquetas de documentos. Cada linha é para um documento.

2. /data/corpus/20ng.txt contém texto bruto de cada documento, cada linha é para a linha correspondente em /data/20ng.txt

3. prepare_data.py é um exemplo para preparar seus próprios dados, observe que ' n' é removido em seus documentos ou frases.

Versão indutiva

Uma versão indutiva do texto GCN é fast_text_gcn, onde os documentos de teste não são incluídos no processo de treinamento.