quanda.utils package

Utilities.

Subpackages

• quanda.utils.datasets package
  • OnDeviceDataset
    • OnDeviceDataset.__init__()
  • Subpackages
    • quanda.utils.datasets.transformed package
      • Submodules
  • Submodules
    • quanda.utils.datasets.dataset_handlers module
      • DatasetHandler
      • HuggingFaceDatasetHandler
      • HuggingFaceSequenceDatasetHandler
      • TorchDatasetHandler
      • get_dataset_handler()
    • quanda.utils.datasets.image_datasets module
      • HFtoTV
    • quanda.utils.datasets.on_device_dataset module
      • OnDeviceDataset
• quanda.utils.functions package
  • cosine_similarity()
  • dot_product_similarity()
  • kendall_rank_corr()
  • spearman_rank_corr()
  • Submodules
    • quanda.utils.functions.correlations module
      • kendall_rank_corr()
      • spearman_rank_corr()
    • quanda.utils.functions.similarities module
      • cosine_similarity()
      • dot_product_similarity()
• quanda.utils.training package
  • BaseTrainer
    • BaseTrainer.fit()
    • BaseTrainer.get_model()
  • BasicLightningModule
    • BasicLightningModule.__init__()
    • BasicLightningModule.configure_optimizers()
    • BasicLightningModule.forward()
    • BasicLightningModule.on_load_checkpoint()
    • BasicLightningModule.on_save_checkpoint()
    • BasicLightningModule.training_step()
    • BasicLightningModule.validation_step()
  • Trainer
    • Trainer.__init__()
    • Trainer.fit()
  • Submodules
    • quanda.utils.training.base_pl_module module
      • BasicLightningModule
    • quanda.utils.training.options module
    • quanda.utils.training.trainer module
      • BaseTrainer
      • Trainer

Submodules

• quanda.utils.cache module
  • BatchedCachedExplanations
    • BatchedCachedExplanations.__init__()
    • BatchedCachedExplanations.keys()
  • Cache
    • Cache.exists()
    • Cache.load()
    • Cache.save()
  • ExplanationsCache
    • ExplanationsCache.exists()
    • ExplanationsCache.load()
    • ExplanationsCache.save()
• quanda.utils.common module
  • DatasetSplit
    • DatasetSplit.__init__()
    • DatasetSplit.exists()
    • DatasetSplit.load()
    • DatasetSplit.save()
    • DatasetSplit.split()
    • DatasetSplit.splits
    • DatasetSplit.to_dict()
  • cache_result()
  • chunked_logits()
  • class_accuracy()
  • default_tensor_type()
  • ds_len()
  • get_load_state_dict_func()
  • get_parent_module_from_name()
  • get_targets()
  • load_last_checkpoint()
  • make_func()
  • map_location_context()
  • move_ds_item_to_device()
  • process_targets()
  • replace_conv1d_with_linear()
  • resolve_config()
  • resolve_device()
  • stable_repr()
  • subsample_dataset()
  • subsample_indices()
• quanda.utils.tasks module
• quanda.utils.tokenization module
  • resolve_tokenizer()
  • tokenize_dataset()