API reference

Data: torchmil.data

• Introduction
• Collate function
• Spatial and sequential representation

Datasets: torchmil.datasets

• Introduction
• Processed MIL Dataset
• Toy Dataset
• CT Scan Dataset
• WSI Dataset
• Binary classification Dataset
• Camelyon16 MIL Dataset
• PANDA MIL Dataset
• RSNA MIL Dataset
• False Frequency Dataset
• Multi-Concept Standard Dataset
• Single Concept Standard Dataset

Modules: torchmil.nn

• Introduction
• Attention
  • Attention Pool
  • Sm Attention Pool
  • Probabilistic Smooth Attention Pool
  • Multihead Self-Attention
  • Multihead Self-Attention with Relative Positional Encoding (iRPE)
  • Nystrom Attention
  • Multihead Cross-Attention
• Graph Neural Networks (GNNs)
  • Deep Graph Convolutional Network (DeepGCN) layer
  • Graph Convolutional Network (GCN) convolution
  • Dense MinCut pooling
• Transformers
  • Transformer base class
  • Conventional Transformer
  • Sm Transformer
  • Nystrom Transformer
  • Transformer with image Relative Positional Encoding (iRPE)
  • Tokens-2-Token
• Sm operator
• Max Pool
• Mean Pool

Models: torchmil.models

• Introduction
• General MIL model
• ABMIL
• CAMIL
• CLAM
• DeepGraphSurv
• DTFDMIL
• DSMIL
• GTP
• IIBMIL
• PatchGCN
• ProbSmoothABMIL
• SmABMIL
• SmTransformerABMIL
• TransformerABMIL
• TransformerProbSmoothABMIL
• TransMIL
• SETMIL
• IIBMIL

Visualize: torchmil.visualize

• Introduction
• Visualizing CT scans
• Visualizing WSI

Utils: torchmil.utils

• Introduction
• Annealing Scheduler
• Graph utils
• Trainer