PyTorch Beam Search Decoding

Pencarian Sinar-PyTorch

Implementasi PyTorch dari decoding pencarian berkas untuk model seq2seq berdasarkan https://github.com/shawnwun/NNDIAL. Penguraian kode dilakukan secara terpisah untuk setiap kalimat dan menyimpan node dalam antrian yang diprioritaskan.

Penggunaan: Anda dapat menentukan hadiah tambahan untuk decoding melalui BeamSearchNode.eval. Berfungsi untuk model dengan dan tanpa perhatian.