trajdl.common.enum module

class trajdl.common.enum.ArrowColName(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

Bases: BaseEnum

DIS_DELTA = 'dis_delta'

ENTITY_ID = 'entity_id'

SEQ = 'seq'

START_TS = 'start_ts'

TS_DELTA = 'ts_delta'

TS_SEQ = 'ts_seq'

class trajdl.common.enum.BaseEnum(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

Bases: Enum

classmethod from_string(value: str)

classmethod parse(value: Any) → BaseEnum

classmethod values() → List[Any]

class trajdl.common.enum.LossEnum(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

Bases: BaseEnum

MEAN = 'mean'

NONE = 'none'

SUM = 'sum'

class trajdl.common.enum.Mode(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

Bases: BaseEnum

EVAL = 'eval'

PRETRAIN = 'pretrain'

TRAIN = 'train'

class trajdl.common.enum.OpenSourceDatasetEnum(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

Bases: BaseEnum

GOWALLA_URL = 'GOWALLA_URL'

PORTO_URL = 'PORTO_URL'

class trajdl.common.enum.ReturnASEnum(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

Bases: BaseEnum

LOCSEQ = 'locseq'

NP = 'np'

PT = 'pt'

PY = 'py'

TRAJ = 'traj'

class trajdl.common.enum.TokenEnum(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

Bases: BaseEnum

BOS_TOKEN = '[BOS]'

EOS_TOKEN = '[EOS]'

MASK_TOKEN = '[MASK]'

PAD_TOKEN = '[PAD]'

UNK_TOKEN = '[UNK]'