Add class name to fit_params

2025-08-16 07:55:54 +00:00 · 2023-02-05 11:29:34 +01:00
parent 75ed3e8f6e
commit 7df037b6f4
1 changed files with 4 additions and 1 deletions
--- 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