Fix eps

aimclub · Oct 30, 2023 · c0a5da2 · c0a5da2
1 parent 316d97d
commit c0a5da2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/examples/tuning_example.py b/examples/tuning_example.py
@@ -68,5 +68,5 @@ def get_search_space():
     # ищем такие параметры, чтобы их сумма была максимальна
     obj_eval = ObjectiveEvaluate(Objective({'sum_metric': ParamsSumMetric.get_value}))
 
-    tuner = IOptTuner(obj_eval, search_space, iterations=10000, n_jobs=1)
+    tuner = IOptTuner(obj_eval, search_space, iterations=10, n_jobs=1)
     tuned_graph = tuner.tune(graph)
diff --git a/golem/core/tuning/iopt_tuner.py b/golem/core/tuning/iopt_tuner.py
@@ -116,7 +116,7 @@ def __init__(self, objective_evaluate: ObjectiveEvaluate,
                  iterations: int = 100,
                  timeout: timedelta = timedelta(minutes=5),
                  n_jobs: int = -1,
-                 eps: float = 0.000001,
+                 eps: float = 0.001,
                  r: float = 2.0,
                  evolvent_density: int = 10,
                  eps_r: float = 0.001,
@@ -137,7 +137,7 @@ def __init__(self, objective_evaluate: ObjectiveEvaluate,
                                                   eps_r=np.double(eps_r),
                                                   refine_solution=refine_solution,
                                                   number_of_parallel_points=determine_n_jobs(n_jobs),
-                                                  timeout=round(timeout.total_seconds()/60))
+                                                  timeout=round(timeout.total_seconds()/60) if self.timeout else -1)
 
     def tune(self, graph: DomainGraphForTune, show_progress: bool = True) -> DomainGraphForTune:
         graph = self.adapter.adapt(graph)