Hi guys, if Bert predicts with only 1 class, there must be something wrong with your trainng pipeline (e.g. check your data and also check your training prediction).
BTW, how about a direct use of NER pretrained weights (may have to use from_pt = True argument), or finetuning from it in your TF code. This should be a great baseline out of the box.

  
      huggingface.co
  
  
    

dslim/bert-large-NER · Hugging Face 10



  
  
    
    
  
  



  
      huggingface.co
  
  
    

dslim/bert-base-NER · Hugging Face 10