Release

cerlymarco · Mar 13, 2022 · 8cce34b · 8cce34b
1 parent 1c00eec
commit 8cce34b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/kerashypetune/_classes.py b/kerashypetune/_classes.py
@@ -340,7 +340,7 @@ def _search(self,
             list(self.folds_scores.values()), axis=0).round(5)
         evaluate = eval_score(mean_score_params)
 
-        self.best_params = [list(f)[evaluate] for f in self.folds_trials.values()]
+        self.best_params = list(self.folds_trials.values())[0][evaluate]
         self.best_params_score = mean_score_params[evaluate]
 
         return self
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 
 HERE = pathlib.Path(__file__).parent
 
-VERSION = '0.2.0'
+VERSION = '0.2.1'
 PACKAGE_NAME = 'keras-hypetune'
 AUTHOR = 'Marco Cerliani'
 AUTHOR_EMAIL = 'cerlymarco@gmail.com'