23 lines
480 B
Python
23 lines
480 B
Python
from enum import Enum
|
|
|
|
|
|
class SpecialToken(Enum):
|
|
# (Enum, str) -> throws an error
|
|
START_TRIPLE_LIST = "<SOTL>"
|
|
START_TRIPLE = "<SOT>"
|
|
END_TRIPLE = "<EOT>"
|
|
SUBJECT = "<SUBJ>"
|
|
RELATIONSHIP = "<PRED>"
|
|
OBJECT = "<OBJ>"
|
|
ABSTRACT = "<ABS>"
|
|
END_OF_SENTENCE = "<EOS>"
|
|
CORPUS_END = "<END>"
|
|
|
|
## Tasks' Token
|
|
RDF_TO_TEXT = "<RDF2TXT>"
|
|
TEXT_TO_RDF = "<TEXT2RDF>"
|
|
CONTINUE_RDF = "<CONTINUERDF>"
|
|
MASK = "<MASK>"
|
|
|
|
# BPE Training:
|