fix bugs

Author: Servon

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
 

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