Text Classification Pytorch

Teks-Klasifikasi-Pytorch

Keterangan

Repositori ini berisi impmentasi berbagai model klasifikasi teks seperti RNN, LSTM, perhatian, CNN, dll dalam kerangka kerja pembelajaran mendalam Pytorch bersama dengan dokumentasi terperinci dari masing -masing model.

Klasifikasi teks adalah salah satu tugas dasar dan terpenting dari pemrosesan bahasa alami. Dalam repositori ini, saya fokus pada salah satu tugas klasifikasi teks seperti itu dan itu adalah analisis sentimen. Sejauh ini saya telah membahas enam model berbeda dalam repo ini.

• Rnn
• LSTM
• LSTM + Perhatian
• Perhatian diri sendiri
• CNN
• Rcnn

Persyaratan

• Python == 3.6.6
• Pytorch == 0.4.0
• TorchText == 0.2.3

Unduhan dan Pengaturan

Setelah Anda mengkloning repo ini, jalankan file Main.py untuk memproses dataset dan untuk melatih model.

$ python main.py

Referensi

• Sebuah kalimat penuh perhatian terstruktur: kertas
• Jaringan Saraf Konvolusional untuk Klasifikasi Kalimat: Kertas
• Jaringan saraf konvolusional berulang untuk klasifikasi teks: kertas

Lisensi

Mit