From 7df037b6f43e7156cba37ef4cb3396cb0fbbb1e8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= Date: Sun, 5 Feb 2023 11:29:34 +0100 Subject: [PATCH] Add class name to fit_params --- benchmark/Experiments.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/benchmark/Experiments.py b/benchmark/Experiments.py index 492686c..a6091f5 100644 --- a/benchmark/Experiments.py +++ b/benchmark/Experiments.py @@ -196,7 +196,10 @@ class Experiment: def _build_fit_params(self, name): if not self.fit_features: return None - res = dict(features=self.datasets.get_features()) + res = dict( + features=self.datasets.get_features(), + class_name=self.datasets.get_class_name(), + ) states = self.datasets.get_states(name) if states is None: return res