2015-10-12 28 views
14

Tôi cần triển khai mạng LSTM hai chiều với lớp CRF ở cuối. Cụ thể là mô hình được trình bày trong bài báo này và đào tạo nó.Triển khai Mạng LSTM-CRF hai hướng

http://www.aclweb.org/anthology/P15-1109

Tôi muốn triển khai tốt nhất bằng Python. Bất cứ ai có thể trình bày một số thư viện hoặc mã mẫu để làm thế nào điều này có thể được thực hiện. Tôi nhìn PyBrain nhưng không hiểu lắm.

Tôi cũng mở bộ công cụ bằng các ngôn ngữ lập trình khác.

+0

Bạn đã hoàn thành công việc của mình và gửi nó đến github? –

+2

https://github.com/chilynn/sequence-labeling –

Trả lời

7

Dưới đây là triển khai mạng LSTM + CRF hai hướng trong TensorFlow: https://github.com/Franck-Dernoncourt/NeuroNER (hoạt động trên Linux/Mac/Windows).

Nó cung cấp các kết quả hiện đại về bộ dữ liệu nhận dạng thực thể được đặt tên.

kiến ​​trúc ANN (nó cũng sử dụng embeddings ký tự):

enter image description here

Như xem trong TensorBoard:

enter image description here

Bạn cũng có thể hình dung embeddings từ:

enter image description here

+0

CRF trong triển khai này ở đâu ?? – Martian2049

+0

@ Matian2049 lớp trên cùng trong ANN. –

0

Đã lâu rồi kể từ khi bạn hỏi, nhưng bây giờ tôi sẽ đề nghị sử dụng anago. Anago được xây dựng trong Keras và bạn có thể thấy mã nguồn nếu bạn cần phải cắt giảm kiến ​​trúc. https://github.com/Hironsan/anago

Các vấn đề liên quan