Exemple #1
0
 def __init__(self, name=None):
     """Responsible for gathering and instantiating experiments, featurizers, and metrics."""
     self.name = name
     self.experiments = [Registry.get_experiment(e) for e in EXPERIMENTS]
     self.featurizers = [Registry.get_featurizer(f)() for f in FEATURIZERS]
     self.metrics = [Registry.get_metric(m)() for m in METRICS]
     self.columns = [
         "Dataset",
         "Featurizer",
         "Experiment",
         "Metric",
         "TrainSize",
         "Sampler",
         "Resampler",
         "Result",
         "TrainResult",
     ]
     self.results = pd.DataFrame(columns=self.columns)
Exemple #2
0
 def __init__(self):
     """
     Responsible for searching featurizer module and importing those specified in config.
     """
     self.featurizers = [Registry.get_featurizer(f)() for f in FEATURIZERS]