abhimishra91 / transformers-tutorials

Github repo with tutorials to fine tune transformers for diff NLP tasks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

NER notebook issue

rajae-Bens opened this issue · comments

Hi,

First, I want to thank u for this great job.
I tried to execute the code of the NER with bert notebook but an error is generated when calculating the f1 score in the validation step

Screen Shot 2020-10-28 at 9 12 02 AM

could someone help plz
thank u

In the valid function, add these lines just before calling F1 Score function.
pred_tags = [pred_tags]
valid_tags = [valid_tags]