textbox.utils.enum_type

class textbox.utils.enum_type.ModelType(value)[source]

Bases: Enum

Type of models.

  • UNCONDITIONAL: Unconditional Generator

  • GAN: Generative Adversarial Net

  • SEQ2SEQ: Seq2Seq Generator

  • ATTRIBUTE: Attribute Generator

ATTRIBUTE = 4
GAN = 2
SEQ2SEQ = 3
UNCONDITIONAL = 1
class textbox.utils.enum_type.SpecialTokens[source]

Bases: object

Special tokens, including PAD, UNK, BOS, EOS. These tokens will by default have token ids 0, 1, 2, 3, respectively.

EOS = '<|endoftext|>'
PAD = '<|pad|>'
SOS = '<|startoftext|>'
UNK = '<|unk|>'