datasets package

Subpackages

• datasets.RTFM package
  • Submodules
  • datasets.RTFM.get_env module
    • create_env()
  • Module contents
• datasets.affect package
  • Submodules
  • datasets.affect.get_bert_embedding module
    • bert_version_data()
    • corresponding_other_modality_ids()
    • get_bert_features()
    • get_rawtext()
    • max_seq_len()
  • datasets.affect.get_data module
    • Affectdataset
      • Affectdataset.__init__()
    • drop_entry()
    • get_dataloader()
    • get_rawtext()
    • z_norm()
  • datasets.affect.get_raw_data module
    • detect_entry_fold()
    • get_audio_visual_text()
    • get_rawtext()
    • get_word2id()
    • get_word_embeddings()
    • glove_embeddings()
    • lpad()
  • Module contents
• datasets.avmnist package
  • Submodules
  • datasets.avmnist.get_data module
    • get_dataloader()
  • Module contents
• datasets.clotho package
  • Submodules
  • datasets.clotho.clotho_data_loader module
    • get_clotho_loader()
  • datasets.clotho.clotho_dataset module
    • ClothoDataset
      • ClothoDataset.__init__()
  • datasets.clotho.collate_fn module
    • clotho_collate_fn()
  • datasets.clotho.get_data module
    • get_dataloaders()
  • Module contents
    • get_clotho_loader()
• datasets.enrico package
  • Submodules
  • datasets.enrico.get_data module
    • EnricoDataset
      • EnricoDataset.__init__()
      • EnricoDataset.featurizeElement()
    • get_dataloader()
  • Module contents
• datasets.gentle_push package
  • Submodules
  • datasets.gentle_push.data_loader module
    • PushTask
      • PushTask.add_dataset_arguments()
      • PushTask.get_dataloader()
      • PushTask.get_dataset_args()
      • PushTask.get_eval_trajectories()
      • PushTask.get_test_trajectories()
      • PushTask.get_train_trajectories()
    • SubsequenceDataset
      • SubsequenceDataset.__init__()
    • TrajectoryNumpy
      • TrajectoryNumpy.controls
      • TrajectoryNumpy.observations
      • TrajectoryNumpy.states
    • split_trajectories()
  • Module contents
• datasets.imdb package
  • Submodules
  • datasets.imdb.get_data module
    • IMDBDataset
      • IMDBDataset.__init__()
    • IMDBDataset_robust
      • IMDBDataset_robust.__init__()
    • get_dataloader()
  • datasets.imdb.vgg module
    • VGGClassifier
      • VGGClassifier.__init__()
      • VGGClassifier.classify()
      • VGGClassifier.get_features()
      • VGGClassifier.resize_and_crop_image()
    • VGGNet
      • VGGNet.__init__()
  • Module contents
• datasets.mimic package
  • Submodules
  • datasets.mimic.get_data module
    • get_dataloader()
  • datasets.mimic.multitask module
    • get_dataloader()
  • Module contents
• datasets.robotics package
  • Submodules
  • datasets.robotics.MultimodalManipulationDataset module
    • MultimodalManipulationDataset
      • MultimodalManipulationDataset.__init__()
    • MultimodalManipulationDataset_robust
      • MultimodalManipulationDataset_robust.__init__()
  • datasets.robotics.MultimodalManipulationDataset_robust module
    • MultimodalManipulationDataset_robust
      • MultimodalManipulationDataset_robust.__init__()
  • datasets.robotics.ProcessForce module
    • ProcessForce
      • ProcessForce.__init__()
  • datasets.robotics.ToTensor module
    • ToTensor
      • ToTensor.__init__()
  • datasets.robotics.get_data module
    • combine_modalitiesbuilder()
    • get_data()
  • datasets.robotics.utils module
    • augment_val()
  • Module contents
• datasets.stocks package
  • Submodules
  • datasets.stocks.get_data module
    • Grouping
      • Grouping.__init__()
      • Grouping.forward()
      • Grouping.training
    • get_dataloader()
  • Module contents

Module contents