/causal_concept_explanation

Primary LanguagePythonApache License 2.0Apache-2.0

causal_concept_explanation

How to use

  • Training and validation dataset preparing
  • Explained model training
  • Explained model evaluation
  • Concept dataset preparing
  • Concept model training (Concept discovered)
  • Causal relationship calculation
  • Causal relationship verification

How to start

conda create -n causal_concept python==3.6.13
conda activate causal_concept
pip install -r requirements.txt

PVR tasks

  1. Dataset Generation for PVR In this stage, two PVR dataset is generated
python data\pvr_dataset_generation.py
  1. Explained Model Training In this stage, we will training 4 model on two dataset including PVR_MNIST and PVR_CIFAR10.
python model_training_for_pvr_task.py
  1. PVR Concept Dataset Generation In this stage, PVR Concept Datasets are generated.
python data\pvr_concepts_dataset_generation.py
  1. Concept Identification In this stage, concept is identified.
python concept_representation_identification.py

Related Works

Thanks for the contributions from the below repos.