Doctorado-ML/benchmark
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/benchmark.git synced 2025-08-17 16:35:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add class name to fit_params

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-02-05 11:29:34 +01:00
parent 75ed3e8f6e
commit 7df037b6f4
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
benchmark/Experiments.py
View File

@@ -196,7 +196,10 @@ class Experiment:
def _build_fit_params(self, name): def _build_fit_params(self, name):
if not self.fit_features: if not self.fit_features:
return None 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) states = self.datasets.get_states(name)
if states is None: if states is None:
return res return res