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

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
explaining_cfg.model.ckpt = "best"
# Setting the path of models folder
explaining_cfg.model.path = "path"
# ----------------------------------------------------------------------- #
# Explainer options
# ----------------------------------------------------------------------- #

View File

@ -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",

View File

@ -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