PyTorch Beam Search Decoding

PyTorch-빔-검색

https://github.com/shawnwun/NNDIAL을 기반으로 seq2seq 모델에 대한 빔 검색 디코딩을 PyTorch로 구현합니다. 디코딩은 각 문장마다 별도로 진행되며 우선 순위가 지정된 대기열에 노드를 저장합니다.

사용법: BeamSearchNode.eval을 통해 디코딩에 대한 추가 보상을 지정할 수 있습니다. 주의가 있거나 없는 모델에 적합합니다.