Abstract base class for converting between text and integers.

A note on padding:

Because text data is typically variable length and nearly always requires padding during training, ID 0 is always reserved for padding. To accommodate this, all TextEncoders behave in certain ways:

  • encode: never returns id 0 (all ids are 1+)
  • decode: drops 0 in the input ids
  • vocab_size: includes ID 0

    New subclasses should be careful to match this behavior.

vocab_size Size of the vocabulary. Decode produces ints [1, vocab_size).



View source

Decodes a list of integers into text.


View source

Encodes text into a list of integers.


View source

Load from file. Inverse of save_to_file.


View source

Store to file. Inverse of load_from_file.