Model Details Model Description This model is electra small finetuned for NER prediction task. The model currently predicts three entities which are given below. 1. Location 2. Person 3. Organization Developed by: விபின் (Vipin) Model type: Google's electra small discriminator Language(s) (NLP): English License: Apache 2.0 Finetuned from model [optional]: Google's electra small discriminator Model Sources [optional] Repository: https://huggingface.co/google/electra small discriminator Uses This model uses tokenizer that is from distilbert family. So the model may predict wrong entities for same word (different sub word). Use 'aggregation strategy' to "max" when using transformer's pipeline. for example 'ashwin ::" ash" = Person win = Location Out of Scope Use May not work well for some long sentences. How to Get Started with the Model Use the code below to get started with the model. Training Details Training Procedure This model is trained for 6 epoch in 3e 4 lr. Evaluation Validation loss is 0.07 for this model
We use cookies for essential functionality and analytics. You can accept or reject analytics cookies.Cookie policy