deepparse model2vec-8m-attention

A deepparse address parser (Seq2Seq, with attention mechanism) using the minishlab/potion-base-8M Model2Vec static embeddings (256 dimensions).

from deepparse.parser import AddressParser

address_parser = AddressParser(model_type="model2vec-8m", attention_mechanism=True)
address_parser("350 rue des Lilas Ouest Québec Québec G1L 1B6")

Install it with pip install deepparse[model2vec].

Training

  • Data: the deepparse/worldwide-addresses dataset (20 countries, 100,000 clean and 50,000 incomplete addresses per country), 90% for training and 10% for validation. Suburb and District are mapped to Municipality, and the County and Country words are removed.
  • Architecture: encoder and decoder hidden size 256, one layer each, with an attention mechanism.
  • Selection: the hidden size (128, 256, 512 or 1024) with the lowest mean validation loss over 5 seeds, then the seed with the lowest validation loss (seed 42, best epoch 26).

Results

Accuracy (%)
Test, this model 98.17
Test, mean ± std over the 5 seeds of this configuration 97.99 ± 0.29
Validation, this model 98.41

The accuracy is the per-word tag accuracy on the test split of the rebuilt worldwide-addresses data.

Limitations

The source data has no Orientation nor GeneralDelivery tags: the model has these outputs but never predicts them (the street orientation, e.g. Ouest or NW, is part of the StreetName). South Korea has almost only incomplete addresses.

Downloads last month
2
Safetensors
Model size
1.19M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train deepparse/model2vec-8m-attention