trajdl.datasets.modules package

1.  Submodules

• trajdl.datasets.modules.abstract module
  • BaseLocSeqDataModule
    • BaseLocSeqDataModule.init_from_parquet()
    • BaseLocSeqDataModule.init_from_table()
  • BaseSeqDataModule
    • BaseSeqDataModule.collate_function()
    • BaseSeqDataModule.init_from_parquet()
    • BaseSeqDataModule.init_from_table()
    • BaseSeqDataModule.num_cpus
    • BaseSeqDataModule.setup()
    • BaseSeqDataModule.test_dataloader()
    • BaseSeqDataModule.test_parquet_path
    • BaseSeqDataModule.test_table
    • BaseSeqDataModule.tokenizer
    • BaseSeqDataModule.train_batch_size
    • BaseSeqDataModule.train_dataloader()
    • BaseSeqDataModule.train_parquet_path
    • BaseSeqDataModule.train_sampler
    • BaseSeqDataModule.train_table
    • BaseSeqDataModule.val_batch_size
    • BaseSeqDataModule.val_dataloader()
    • BaseSeqDataModule.val_parquet_path
    • BaseSeqDataModule.val_sampler
    • BaseSeqDataModule.val_table
  • BaseTrajectoryDataModule
    • BaseTrajectoryDataModule.init_from_parquet()
    • BaseTrajectoryDataModule.init_from_table()
• trajdl.datasets.modules.ctle module
  • CTLEDataModule
    • CTLEDataModule.collate_function()
    • CTLEDataModule.mask_prob
• trajdl.datasets.modules.gmvsae module
  • GMVSAEDataModule
    • GMVSAEDataModule.collate_function()
    • GMVSAEDataModule.num_train_batches
    • GMVSAEDataModule.num_train_buckets
    • GMVSAEDataModule.num_val_batches
    • GMVSAEDataModule.num_val_buckets
    • GMVSAEDataModule.setup()
• trajdl.datasets.modules.hier module
  • HIERDataModule
    • HIERDataModule.collate_function()
• trajdl.datasets.modules.stlstm module
  • HSTLSTMDataModule
    • HSTLSTMDataModule.collate_function()
    • HSTLSTMDataModule.num_train_batches
    • HSTLSTMDataModule.num_train_buckets
    • HSTLSTMDataModule.num_val_batches
    • HSTLSTMDataModule.num_val_buckets
    • HSTLSTMDataModule.setup()
  • STLSTMDataModule
    • STLSTMDataModule.collate_function()
    • STLSTMDataModule.loc_bucketizer
    • STLSTMDataModule.num_train_batches
    • STLSTMDataModule.num_train_buckets
    • STLSTMDataModule.num_val_batches
    • STLSTMDataModule.num_val_buckets
    • STLSTMDataModule.setup()
    • STLSTMDataModule.ts_bucketizer
• trajdl.datasets.modules.t2vec module
  • T2VECDataModule
    • T2VECDataModule.collate_fn_train()
    • T2VECDataModule.setup()
    • T2VECDataModule.train_dataloader()
    • T2VECDataModule.val_dataloader()
  • T2VECDataModuleV2
    • T2VECDataModuleV2.collate_function()
    • T2VECDataModuleV2.k
    • T2VECDataModuleV2.num_train_batches
    • T2VECDataModuleV2.num_train_buckets
    • T2VECDataModuleV2.num_val_batches
    • T2VECDataModuleV2.num_val_buckets
    • T2VECDataModuleV2.setup()
  • downsampling_distort()
  • generate_samples()
• trajdl.datasets.modules.tuler module
  • TULERDataModule
    • TULERDataModule.collate_function()
    • TULERDataModule.setup()
    • TULERDataModule.user_map