ontology-transfer
This model is a fine-tuned version of
scales-okn/docket-language-model on the None dataset.
It achieves the following results on the evaluation set:
- Loss: 0.0462
- Accuracy: 0.9952
- F1: 0.9926
- Precision: 0.9853
- Recall: 1.0
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 3e-05
- train_batch_size: 4
- eval_batch_size: 8
- seed: 42
- gradient_accumulation_steps: 2
- total_train_batch_size: 8
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- lr_scheduler_warmup_ratio: 0.06
- num_epochs: 5
Training results
| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
|---|
| 0.1769 | 0.67 | 100 | 0.1263 | 0.9571 | 0.9362 | 0.8919 | 0.9851 |
| 0.1662 | 1.34 | 200 | 0.0640 | 0.9857 | 0.9778 | 0.9706 | 0.9851 |
| 0.1037 | 2.01 | 300 | 0.0927 | 0.9857 | 0.9781 | 0.9571 | 1.0 |
| 0.0003 | 2.68 | 400 | 0.0936 | 0.9810 | 0.9710 | 0.9437 | 1.0 |
| 0.0002 | 3.36 | 500 | 0.0462 | 0.9952 | 0.9926 | 0.9853 | 1.0 |
| 0.0001 | 4.03 | 600 | 0.0694 | 0.9857 | 0.9781 | 0.9571 | 1.0 |
| 0.0001 | 4.7 | 700 | 0.0692 | 0.9857 | 0.9781 | 0.9571 | 1.0 |
Framework versions
- Transformers 4.23.1
- Pytorch 1.13.0+cu117
- Datasets 2.6.1
- Tokenizers 0.13.1
Public release information
This model is released by the SCALES Open Knowledge Network under the GNU General
Public License v3.0. It is derived from scales-okn/docket-language-model and is
intended for research and development involving legal-document classification or
information extraction. It is not legal advice.
The organization has reviewed the release decision and confirmed that the model's
training data and resulting weights are legally and ethically releasable. Users are
responsible for evaluating accuracy, bias, privacy, and fitness for their own use.
The repository includes PyTorch .bin artifacts. Hugging Face's server-side security
scan reported no file issues before publication. As with any serialized model
artifact, load it only with maintained libraries and in an appropriately isolated
environment.
