trax icon indicating copy to clipboard operation
trax copied to clipboard

How to implement NER model using trax

Open omerarshad opened this issue 5 years ago • 5 comments

How to use Trax to implement NER model in NLP using reformer.

omerarshad avatar Jan 28 '20 13:01 omerarshad

We're working on providing more examples, will update here when there's one for NER.

lukaszkaiser avatar Apr 28 '20 21:04 lukaszkaiser

Hey, I've been working on this colab notebook for implementing NER using Reformer.

Please go through it and tell suggestions.

SauravMaheshkar avatar Oct 19 '20 18:10 SauravMaheshkar

On training for 100 steps, using kaggle kernels I got an accuracy of 85%.

You can find the notebook here

SauravMaheshkar avatar Oct 20 '20 07:10 SauravMaheshkar

I've had a go with that notebook but I can't figure out how to use this model to do NER on a new sentence. (I've posted question on stackexchange: https://ai.stackexchange.com/questions/25442/using-reformer-model-to-do-ner-in-trax)

One thing that puzzles me about the 85% accuracy on this test. Given the large proportion of tags that are 'O', is 85% accuracy really that good? The LSTM example that Coursera uses achieves 90-95% accuracy, and I would have hoped that Reformer is better.

alanbuxton avatar Jan 01 '21 11:01 alanbuxton

@SauravMaheshkar could you please offer some help regarding predictions using the NER_Reformer code that you have demonstrated ? You help will be highly appreciated.

ghost avatar May 12 '22 13:05 ghost