maia2000/ner-camembert

# CamemBERT Model - NEEDS RETRAINING

Current Status: OUTDATED

This model was trained with CamembertTokenizer (old) but the code now uses CamembertTokenizerFast (required for NER).

Symptoms:

• Wrong entity predictions (Paris PER instead of DEP)
• "Departure: Not found" even when cities are mentioned
• Mixed up entity types (VIA, MISC instead of DEP, ARR)

Fix: Retrain the Model

Option 1: Using Notebook (Recommended) Open: notebooks/Complete_NER_Training_and_Analysis.ipynb Run: Cell 16 (Part 3.2 - Train CamemBERT) Time: ~10-15 min (GPU) or ~45 min (CPU)

Option 2: Python Script python from src.train_camembert import CamemBERTNERTrainer trainer = CamemBERTNERTrainer("camembert-base", "models/camembert", 128) trainer.train("data/train.csv", "data/val.csv", epochs=3, batch_size=16)

After Retraining:

`ash python app.py --model models/camembert --port 5000

Should now correctly predict: "Paris Lyon" DEP: Paris, ARR: Lyon

`

---

Last Updated: 2026-02-02