diff --git a/explaining_framework/config/explaining_config.py b/explaining_framework/config/explaining_config.py index 592e45d..b14db35 100644 --- a/explaining_framework/config/explaining_config.py +++ b/explaining_framework/config/explaining_config.py @@ -70,6 +70,9 @@ def set_cfg(explaining_cfg): # Set wether or not load the best model for given dataset or a path explaining_cfg.model.ckpt = "best" + # Setting the path of models folder + explaining_cfg.model.path = "path" + # ----------------------------------------------------------------------- # # Explainer options # ----------------------------------------------------------------------- # diff --git a/explaining_framework/utils/explaining/cmd_args.py b/explaining_framework/utils/explaining/cmd_args.py index fabd74e..23840fc 100644 --- a/explaining_framework/utils/explaining/cmd_args.py +++ b/explaining_framework/utils/explaining/cmd_args.py @@ -3,7 +3,7 @@ import argparse def parse_args() -> argparse.Namespace: r"""Parses the command line arguments.""" - parser = argparse.ArgumentParser(description="GraphGym") + parser = argparse.ArgumentParser(description="ExplainingFramework") parser.add_argument( "--cfg", diff --git a/explaining_framework/utils/explaining/explaining_exp.py b/explaining_framework/utils/explaining/explaining_exp.py index 84b8413..c01a5db 100644 --- a/explaining_framework/utils/explaining/explaining_exp.py +++ b/explaining_framework/utils/explaining/explaining_exp.py @@ -1,5 +1,14 @@ -class Explaining(object): - def __init__(self, cfg: dict, explaining_cfg: dict, explainer_cfg: dict = None): - self.cfg = cfg +class ExplainingOutline(object): + def __init__(self, explaining_cfg: CN, explainer_cfg: CN = None): self.explaining_cfg = explaining_cfg self.explainer_cfg = explainer_cfg + + + + def load_cfg(self): + if self + + + + +