Food NER
Github Repo: https://github.com/randymi01/food_ner
Spacy Food Name Entity Recognition (NER) model trained on StanfordNLP CRF recipe dataset
Installation
Use the package manager pip to install spacy version spacy==3.5.0 and then download the spacy en_core_web_sm model.
pip install spacy==3.5.0
python -m spacy download en_core_web_sm
Usage
import spacy
nlp = spacy.load("model")
# returns (spring mix, chicken breast, chili, hamburger meat)
nlp("I have spring mix, chicken breast, chili, and hamburger meat").ents
Model Hyperparameters
- Epochs: 10
- Batch Size: 4-32
- Optimizer: Adam
- lr = 5e-03
- drop_rate = 0.5
Model Performance
License
- Downloads last month
- 0