Edit model card

Food NER

Github Repo: https://github.com/randymi01/food_ner

Spacy Food Name Entity Recognition (NER) model trained on StanfordNLP CRF recipe dataset


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


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

alt text alt text



Downloads last month
Unable to determine this model’s pipeline type. Check the docs .