araison/explaining_framework
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Reformating, fixing

Browse source
This commit is contained in:
araison 2023-01-04 10:41:34 +01:00
parent d9628ff947
commit 7e32d6fd3a
3 changed files with 145 additions and 110 deletions
Download patch file Download diff file Expand all files Collapse all files

27
explaining_framework/config/explaining_config.py
View file

@ -57,7 +57,7 @@ def set_cfg(explaining_cfg):
explaining_cfg.dataset.name = "Cora"
explaining_cfg.dataset.specific_items = None
explaining_cfg.dataset.items = None
explaining_cfg.run_topological_stat = True
@ -110,26 +110,33 @@ def set_cfg(explaining_cfg):
# Thresholding options
# ----------------------------------------------------------------------- #
explaining_cfg.threshold_config = CN()
explaining_cfg.threshold = CN()
explaining_cfg.threshold_config.threshold_type = None
explaining_cfg.threshold.config = CN()
explaining_cfg.threshold.config.type = "all"
explaining_cfg.threshold_config.value = [i * 0.05 for i in range(21)]
explaining_cfg.threshold_config.relu_and_normalize = True
# Select device: 'cpu', 'cuda', 'auto'
explaining_cfg.threshold.value = CN()
explaining_cfg.threshold.value.hard = [i * 0.05 for i in range(21)]
explaining_cfg.threshold.value.topk = [2, 3, 5, 10, 20, 30, 50]
# which objectives metrics to computes, either all or one in particular if implemented
explaining_cfg.metrics = CN()
explaining_cfg.metrics.name = "all"
explaining_cfg.metrics.sparsity = CN()
explaining_cfg.metrics.sparsity.name = "all"
explaining_cfg.metrics.fidelity = CN()
explaining_cfg.metrics.fidelity.name = "all"
explaining_cfg.metrics.accuracy = CN()
explaining_cfg.metrics.accuracy.name = "all"
# Whether or not recomputing metrics if they already exist
explaining_cfg.metrics.force = False
explaining_cfg.adjust = CN()
explaining_cfg.adjust.strategy = "rpn"
explaining_cfg.attack = CN()
explaining_cfg.attack.name = "all"
# Select device: 'cpu', 'cuda', 'auto'
explaining_cfg.accelerator = "auto"