Text Classification Pytorch

Text-Classification-Pytorch

説明

このリポジトリには、Pytorch Deep LearningフレームワークにおけるRNN、LSTM、注意、CNNなど、各モデルの詳細なドキュメントに加えて、RNN、LSTM、注意、CNNなどのさまざまなテキスト分類モデルの実装が含まれています。

テキスト分類は、自然言語処理の基本的かつ最も重要なタスクの1つです。このリポジトリでは、私はそのようなテキスト分類タスクの1つに焦点を当てており、それがセンチメント分析です。これまで、このレポで次の6つの異なるモデルをカバーしてきました。

• RNN
• LSTM
• LSTM +注意
• 自己注意
• CNN
• rcnn

要件

• python == 3.6.6
• pytorch == 0.4.0
• torchtext == 0.2.3

ダウンロードとセットアップ

このレポをクローンしたら、Main.pyファイルを実行してデータセットを処理し、モデルをトレーニングします。

$ python main.py

参照

• 構造化された自己完全な文の埋め込み：紙
• 文の分類のための畳み込みニューラルネットワーク：論文
• テキスト分類のための再発畳み込みニューラルネットワーク：論文

ライセンス

mit