ybracke's picture
Update README.md
c4f2c13 verified
|
raw
history blame
3.52 kB
metadata
license: apache-2.0
base_model: google/byt5-small
tags:
  - generated_from_trainer
language: de
model-index:
  - name: ybracke/transnormer-19c-beta-v02
    results:
      - task:
          name: Historic Text Normalization
          type: translation
        dataset:
          name: DTA EvalCorpus
          type: text
          split: test
        metrics:
          - name: Word Accuracy
            type: accuracy
            value: 0.98878
          - name: Word Accuracy (case insensitive)
            type: accuracy
            value: 0.99343
pipeline_tag: text2text-generation
library_name: transformers

Transnormer 19th century (beta v02)

This model generates a normalized version of historical input text for German from the 19th (and late 18th) century.
The base model google/byt5-small was fine-tuned on a modified version of the DTA EvalCorpus (see section Training and evaluation data).

Model description

Demo Usage

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("ybracke/transnormer-19c-beta-v02")
model = AutoModelForSeq2SeqLM.from_pretrained("ybracke/transnormer-19c-beta-v02")
sentence = "Die Königinn ſaß auf des Pallaſtes mittlerer Tribune."
inputs = tokenizer(sentence, return_tensors="pt",)
outputs = model.generate(**inputs, num_beams=4, max_length=128)
print(tokenizer.batch_decode(outputs, skip_special_tokens=True))
# >>> ['Die Königin saß auf des Palastes mittlerer Tribüne.']

Or use this model with the pipeline API like this:

from transformers import pipeline
transnormer = pipeline('text2text-generation', model='ybracke/transnormer-19c-beta-v02')
sentence = "Die Königinn ſaß auf des Pallaſtes mittlerer Tribune."
print(transnormer(sentence))
# >>> [{'generated_text': 'Die Königin saß auf des Palastes mittlerer Tribüne.'}]

Training and evaluation data

The model was fine-tuned and evaluated on splits derived from the DTA EvalCorpus, a parallel corpus containing of 121 texts from Deutsches Textarchiv (German Text Archive). The corpus was originally created by aligning historic prints in original spelling with an edition in contemporary orthography.

The original corpus creators applied some corrections to the modern versions (see Jurish et al. 2013). For our use of the corpus, we further improved the quality of the normalized part of the corpus by enforcing spellings that accord to the German orthography reform (post 1996) and by applying selected rules of the LanguageTool and custom replacements to remove some errors and inconsistencies. We plan to publish the corpus as a dataset on the Huggingface Hub in the future.

The training set contains 96 documents with 4.6M source tokens, the dev and test set contain 13 documents (405K tokens) and 12 documents (381K tokens), respectively.

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 5e-05
  • train_batch_size: 8
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 10 (published model: 8 epochs)

Framework versions

  • Transformers 4.31.0
  • Pytorch 2.1.0+cu121
  • Datasets 2.18.0
  • Tokenizers 0.13.3