Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
Torch Concepts
PyC Logo

Usage

  • Installation
  • User Guide
    • Interpretable Layers and Interventions
    • Interpretable Probabilistic Models
    • Structural Equation Models
    • Out-of-the-box Models
    • Conceptarium
  • Contributing Guide
  • License

API Reference

  • Low-level API
    • Base classes (low level)
      • torch_concepts.nn.BaseConceptLayer
      • torch_concepts.nn.BaseEncoder
      • torch_concepts.nn.BasePredictor
      • torch_concepts.nn.BaseGraphLearner
      • torch_concepts.nn.BaseInference
      • torch_concepts.nn.BaseIntervention
    • Concept Encoders
      • torch_concepts.nn.LinearZC
      • torch_concepts.nn.LinearUC
      • torch_concepts.nn.StochasticZC
      • torch_concepts.nn.LinearZU
      • torch_concepts.nn.SelectorZU
    • Concept Predictors
      • torch_concepts.nn.LinearCC
      • torch_concepts.nn.MixCUC
      • torch_concepts.nn.HyperLinearCUC
      • torch_concepts.nn.CallableCC
    • Intervention Strategies and Context Manager
      • torch_concepts.nn.RewiringIntervention
      • torch_concepts.nn.GroundTruthIntervention
      • torch_concepts.nn.DoIntervention
      • torch_concepts.nn.DistributionIntervention
      • torch_concepts.nn.intervention
    • Intervention Policies
      • torch_concepts.nn.UniformPolicy
      • torch_concepts.nn.RandomPolicy
      • torch_concepts.nn.UncertaintyInterventionPolicy
    • Graph Learners
      • torch_concepts.nn.WANDAGraphLearner
    • Dense Layers
      • torch_concepts.nn.Dense
      • torch_concepts.nn.MLP
      • torch_concepts.nn.ResidualMLP
  • Mid-level API
    • Base classes (mid level)
      • torch_concepts.nn.BaseConstructor
    • Random Variables
      • torch_concepts.Variable
      • torch_concepts.EndogenousVariable
      • torch_concepts.ExogenousVariable
      • torch_concepts.InputVariable
    • Probabilistic Models
      • torch_concepts.nn.ProbabilisticModel
      • torch_concepts.nn.ParametricCPD
      • torch_concepts.nn.BipartiteModel
      • torch_concepts.nn.GraphModel
    • Probabilistic Inference
      • torch_concepts.nn.ForwardInference
      • torch_concepts.nn.DeterministicInference
      • torch_concepts.nn.AncestralSamplingInference
    • Model Constructors
      • torch_concepts.nn.BipartiteModel
      • torch_concepts.nn.GraphModel
  • High-level API
    • Base classes (high level)
      • torch_concepts.nn.BaseModel
    • Annotations
      • torch_concepts.annotations.AxisAnnotation
      • torch_concepts.annotations.Annotations
    • High-Level Models
      • torch_concepts.nn.ConceptBottleneckModel
      • torch_concepts.nn.ConceptBottleneckModel_Joint
      • torch_concepts.nn.BlackBox
    • Loss Functions
      • torch_concepts.nn.modules.loss.ConceptLoss
      • torch_concepts.nn.modules.loss.WeightedConceptLoss
    • Metrics
      • torch_concepts.nn.modules.metrics.ConceptMetrics
      • torch_concepts.nn.functional.completeness_score
      • torch_concepts.nn.functional.intervention_score
      • torch_concepts.nn.functional.cace_score
  • Loss Functions
    • torch_concepts.nn.modules.loss.ConceptLoss
    • torch_concepts.nn.modules.loss.WeightedConceptLoss
  • Metrics
    • torch_concepts.nn.modules.metrics.ConceptMetrics
    • torch_concepts.nn.functional.completeness_score
    • torch_concepts.nn.functional.intervention_score
    • torch_concepts.nn.functional.cace_score
  • Functional API
    • torch_concepts.nn.functional.grouped_concept_exogenous_mixture
    • torch_concepts.nn.functional.selection_eval
    • torch_concepts.nn.functional.confidence_selection
    • torch_concepts.nn.functional.soft_select
    • torch_concepts.nn.functional.linear_equation_eval
    • torch_concepts.nn.functional.linear_equation_expl
    • torch_concepts.nn.functional.logic_rule_eval
    • torch_concepts.nn.functional.logic_memory_reconstruction
    • torch_concepts.nn.functional.logic_rule_explanations
    • torch_concepts.nn.functional.completeness_score
    • torch_concepts.nn.functional.intervention_score
    • torch_concepts.nn.functional.cace_score
    • torch_concepts.nn.functional.residual_concept_causal_effect
    • torch_concepts.nn.functional.selective_calibration
    • torch_concepts.nn.functional.edge_type
    • torch_concepts.nn.functional.prune_linear_layer
  • Data
    • Data Base Classes
      • torch_concepts.data.base.dataset.ConceptDataset
      • torch_concepts.data.base.datamodule.ConceptDataModule
      • torch_concepts.data.base.scaler.Scaler
      • torch_concepts.data.base.splitter.Splitter
    • Data Modules
      • torch_concepts.data.datamodules.BnLearnDataModule
    • Datasets
      • torch_concepts.data.datasets.bnlearn.BnLearnDataset
      • torch_concepts.data.datasets.toy.ToyDataset
      • torch_concepts.data.datasets.toy.CompletenessDataset
      • torch_concepts.data.datasets.mnist.ColorMNISTDataset
      • torch_concepts.data.datasets.mnist.MNISTAddition
      • torch_concepts.data.datasets.mnist.PartialMNISTAddition
      • torch_concepts.data.datasets.mnist.MNISTEvenOdd
      • torch_concepts.data.datasets.celeba.CelebADataset
      • torch_concepts.data.datasets.cub.CUBDataset
      • torch_concepts.data.datasets.awa2.AwA2Dataset
      • torch_concepts.data.datasets.cebab.CEBaBDataset
      • torch_concepts.data.datasets.traffic.TrafficLights
    • Preprocessing
      • torch_concepts.data.preprocessing.autoencoder.SimpleAutoencoder
      • torch_concepts.data.preprocessing.autoencoder.AutoencoderTrainer
      • torch_concepts.data.preprocessing.autoencoder.extract_embs_from_autoencoder
    • Scalers
      • torch_concepts.data.scalers.standard.StandardScaler
    • Data Splitters
      • torch_concepts.data.splitters.random.RandomSplitter
      • torch_concepts.data.splitters.coloring.ColoringSplitter
    • Backbone Networks
    • Data I/O
      • torch_concepts.data.io.extract_zip
      • torch_concepts.data.io.extract_tar
      • torch_concepts.data.io.save_pickle
      • torch_concepts.data.io.load_pickle
      • torch_concepts.data.io.download_url
      • torch_concepts.data.io.DownloadProgressBar
    • Data Utilities
      • torch_concepts.data.utils.ensure_list
      • torch_concepts.data.utils.files_exist
      • torch_concepts.data.utils.parse_tensor
      • torch_concepts.data.utils.convert_precision
      • torch_concepts.data.utils.colorize
      • torch_concepts.data.utils.affine_transform
      • torch_concepts.data.utils.transform_images
      • torch_concepts.data.utils.assign_random_values
      • torch_concepts.data.utils.assign_values_based_on_intervals
      • torch_concepts.data.utils.colorize_and_transform
  • Distributions
    • torch_concepts.distributions.Delta

Indices

  • Index
  • Module Index
Back to top
View this page

DataΒΆ

Data APIs provide utilities for loading, preprocessing, and managing datasets.

  • Data Base Classes
  • Data Modules
  • Datasets
  • Preprocessing
  • Scalers
  • Data Splitters
  • Backbone Networks
  • Data I/O
  • Data Utilities
Next
Data Base Classes
Previous
torch_concepts.nn.functional.prune_linear_layer
Copyright © 2026, PyC Team
Made with Sphinx and @pradyunsg's Furo