farsi_lastname_classifier_bert
This model is trained to classify Iranian last names.
To use it, type a last name in the space provided on the right and then click on "compute".
The model computes probability of the last name being Persian.
The compute takes a few seconds to load for the first try (because it needs to load the model first). Subsequent attempt should take only milliseconds.
In practice the model can compute the results for an entire batch of data (last names) in a fraction of a second.
It achieves the following results on the evaluation set:
- Loss: 0.0863
- Accuracy: 0.976
Model description
Model is based on Bert ("bert-base-cased")
The following hyperparameters were used during training:
- learning_rate: 1e-05
- train_batch_size: 128
- eval_batch_size: 256
- seed: 42
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: cosine
- lr_scheduler_warmup_ratio: 0.1
- num_epochs: 10
- mixed_precision_training: Native AMP
Training results
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|---|
| No log | 1.0 | 12 | 0.6325 | 0.588 |
| No log | 2.0 | 24 | 0.3414 | 0.952 |
| No log | 3.0 | 36 | 0.2496 | 0.97 |
| No log | 4.0 | 48 | 0.1674 | 0.976 |
| No log | 5.0 | 60 | 0.1160 | 0.976 |
| No log | 6.0 | 72 | 0.0917 | 0.972 |
| No log | 7.0 | 84 | 0.0896 | 0.974 |
| No log | 8.0 | 96 | 0.0874 | 0.974 |
| No log | 9.0 | 108 | 0.0869 | 0.974 |
| No log | 10.0 | 120 | 0.0863 | 0.976 |
Framework versions
- Transformers 4.24.0
- Pytorch 1.12.1+cu113
- Datasets 2.6.1
- Tokenizers 0.13.2
