adding prediction function

Hironsan / anago

Bidirectional LSTM-CRF and ELMo for Named-Entity Recognition, Part-of-Speech Tagging and so on.

https://anago.herokuapp.com/

MIT License

1.48k stars 371 forks source link

adding prediction function #83

Closed yeger00 closed 5 years ago

yeger00 commented 5 years ago

I wrote a prediction function that return the predictions. IT is useful in case the score() function returns low f1 value, and you need to analyze the predictions in order to understand how to make the model better. Let me know if you have any comments, or if I used the library in a wrong way.

Thanks,