text_gcn

텍스트 _gcn

본 논문에서 텍스트 GCN의 구현 :

Liang Yao, Chengsheng Mao, Yuan Luo. "텍스트 분류를위한 그래프 컨볼 루션 네트워크." 인공 지능에 관한 33 번째 AAAI 회의 (AAAI-19), 7370-7377

필요하다

파이썬 2.7 또는 3.6

텐서 플로우> = 1.4.0

결과를 재현합니다

1. python remove_words.py 20ng 실행하십시오

2. python build_graph.py 20ng 실행하십시오

3. python train.py 20ng 실행하십시오

4. 다른 데이터 세트에 대한 결과를 생성 할 때 3 개 이상의 명령 줄에서 20ng 3 개의 명령 줄을 R8 , R52 , ohsumed 및 mr 로 변경하십시오.

입력 데이터 예

1. /data/20ng.txt 문서 이름, 교육/테스트 분할, 문서 레이블을 나타냅니다. 각 줄은 문서 용입니다.

2. /data/corpus/20ng.txt 각 문서의 원시 텍스트가 포함되어 있으며 각 줄은 /data/20ng.txt 의 해당 줄에 대한 것입니다.

3. prepare_data.py 자신의 데이터를 준비하기위한 예입니다. 문서 나 문장에서 ' n'이 제거됩니다.

유도 버전

텍스트 GCN의 귀납적 버전은 FAST_TEXT_GCN이며, 여기서 테스트 문서가 교육 프로세스에 포함되지 않습니다.