trajdl.common package

1.  Submodules

• trajdl.common.enum module
  • ArrowColName
    • ArrowColName.DIS_DELTA
    • ArrowColName.ENTITY_ID
    • ArrowColName.SEQ
    • ArrowColName.START_TS
    • ArrowColName.TS_DELTA
    • ArrowColName.TS_SEQ
  • BaseEnum
    • BaseEnum.from_string()
    • BaseEnum.parse()
    • BaseEnum.values()
  • LossEnum
    • LossEnum.MEAN
    • LossEnum.NONE
    • LossEnum.SUM
  • Mode
    • Mode.EVAL
    • Mode.PRETRAIN
    • Mode.TRAIN
  • OpenSourceDatasetEnum
    • OpenSourceDatasetEnum.GOWALLA_URL
    • OpenSourceDatasetEnum.PORTO_URL
  • ReturnASEnum
    • ReturnASEnum.LOCSEQ
    • ReturnASEnum.NP
    • ReturnASEnum.PT
    • ReturnASEnum.PY
    • ReturnASEnum.TRAJ
  • TokenEnum
    • TokenEnum.BOS_TOKEN
    • TokenEnum.EOS_TOKEN
    • TokenEnum.MASK_TOKEN
    • TokenEnum.PAD_TOKEN
    • TokenEnum.UNK_TOKEN
• trajdl.common.samples module
  • GMVSAESample
    • GMVSAESample.batch_size
    • GMVSAESample.decoder_labels
    • GMVSAESample.decoder_lengths
    • GMVSAESample.decoder_seq
    • GMVSAESample.encoder_lengths
    • GMVSAESample.encoder_seq
    • GMVSAESample.mask
  • STLSTMSample
    • STLSTMSample.batch_size
    • STLSTMSample.labels
    • STLSTMSample.loc_seq
    • STLSTMSample.mask
    • STLSTMSample.sd_lower_seq
    • STLSTMSample.sd_upper_seq
    • STLSTMSample.td_lower_seq
    • STLSTMSample.td_upper_seq
    • STLSTMSample.valid_lengths
  • T2VECSample
    • T2VECSample.batch_size
    • T2VECSample.lengths
    • T2VECSample.src
    • T2VECSample.target
  • TULERSample
    • TULERSample.batch_size
    • TULERSample.labels
    • TULERSample.seq_len
    • TULERSample.src