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

New features

This commit is contained in:
araison 2022-12-27 11:25:30 +01:00
parent 9397934825
commit d52bc702a2
3 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
explaining_framework/config/explaining_config.py
View File

@ -70,6 +70,9 @@ def set_cfg(explaining_cfg):
# Set wether or not load the best model for given dataset or a path # Set wether or not load the best model for given dataset or a path
explaining_cfg.model.ckpt = "best" explaining_cfg.model.ckpt = "best"
# Setting the path of models folder
explaining_cfg.model.path = "path"
# ----------------------------------------------------------------------- # # ----------------------------------------------------------------------- #
# Explainer options # Explainer options
# ----------------------------------------------------------------------- # # ----------------------------------------------------------------------- #

2
explaining_framework/utils/explaining/cmd_args.py
View File

@ -3,7 +3,7 @@ import argparse
def parse_args() -> argparse.Namespace: def parse_args() -> argparse.Namespace:
r"""Parses the command line arguments.""" r"""Parses the command line arguments."""
parser = argparse.ArgumentParser(description="GraphGym") parser = argparse.ArgumentParser(description="ExplainingFramework")
parser.add_argument( parser.add_argument(
"--cfg", "--cfg",

15
explaining_framework/utils/explaining/explaining_exp.py
View File

@ -1,5 +1,14 @@
class Explaining(object): class ExplainingOutline(object):
def __init__(self, cfg: dict, explaining_cfg: dict, explainer_cfg: dict = None): def __init__(self, explaining_cfg: CN, explainer_cfg: CN = None):
self.cfg = cfg
self.explaining_cfg = explaining_cfg self.explaining_cfg = explaining_cfg
self.explainer_cfg = explainer_cfg self.explainer_cfg = explainer_cfg
def load_cfg(self):
if self