From d52bc702a23517cc5832ff642fdce5fe245b4b76 Mon Sep 17 00:00:00 2001 From: araison Date: Tue, 27 Dec 2022 11:25:30 +0100 Subject: [PATCH] New features --- explaining_framework/config/explaining_config.py | 3 +++ explaining_framework/utils/explaining/cmd_args.py | 2 +- .../utils/explaining/explaining_exp.py | 15 ++++++++++++--- 3 files changed, 16 insertions(+), 4 deletions(-) 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 + + + + +