text_gcn

text_gcn

私たちの論文でのテキストGCNの実装：

Liang Yao、Chengsheng Mao、Yuan Luo。 「テキスト分類のためのグラフ畳み込みネットワーク。」人工知能に関する第33回AAAI会議（AAAI-19）、7370-7377

必要とする

Python 2.7または3.6

Tensorflow> = 1.4.0

結果の再現

1. python remove_words.py 20ngを実行します

2. python build_graph.py 20ngを実行します

3. python train.py 20ng実行します

4. 他のデータセットの結果を生成する際に、上記の3つのコマンドラインでR8 、 R52 、 ohsumed 、 mrに20ng変更します。

インプットデータの例

1. /data/20ng.txtドキュメント名、トレーニング/テストの分割、ドキュメントラベルを示します。各行はドキュメント用です。

2. /data/corpus/20ng.txt各ドキュメントの生のテキストが含まれています。各行は/data/20ng.txtの対応する行用です

3. prepare_data.py 、独自のデータを準備するための例です。「 n」はドキュメントまたは文で削除されていることに注意してください。

帰納バージョン

テキストGCNの誘導バージョンはfast_text_gcnであり、テストドキュメントはトレーニングプロセスに含まれていません。