fix bugs

polixir · Feb 23, 2021 · 91a31d4 · 91a31d4
1 parent 9086d73
commit 91a31d4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/zoopt/algos/opt_algorithms/racos/racos_common.py b/zoopt/algos/opt_algorithms/racos/racos_common.py
@@ -191,9 +191,8 @@ def selection(self):
 
         new_data = sorted(self._data, key=lambda x: x.get_value())
         self._data = new_data[0: self._parameter.get_train_size()]
-        self._positive_data = new_data[0: self._parameter.get_positive_size()]
-        self._negative_data = new_data[
-            self._parameter.get_positive_size(): self._parameter.get_train_size()]
+        self._positive_data = self._data[0: self._parameter.get_positive_size()]
+        self._negative_data = self._data[self._parameter.get_positive_size():]
         self._best_solution = self._positive_data[0]
         return
 

diff --git a/zoopt/algos/opt_algorithms/racos/sracos.py b/zoopt/algos/opt_algorithms/racos/sracos.py
@@ -289,9 +289,8 @@ def complete(self, solution, result):
         solution.set_value(result)
         if self.complete_num < self._parameter.get_train_size():
             self._data.append(solution)
-        elif self.complete_num == self._parameter.get_train_size():
-            self._data.append(solution)
             self.selection()
+        elif self.complete_num == self._parameter.get_train_size():
             best_solution_so_far = self.update_classifier(solution)
             self.semaphore = 1
             return best_solution_so_far