Przeglądaj źródła

update number of sample used for surrogate

Jérôme BUISINE 3 lat temu
rodzic
commit
2144b1d5da
1 zmienionych plików z 1 dodań i 1 usunięć
  1. 1 1
      optimization/ILSSurrogate.py

+ 1 - 1
optimization/ILSSurrogate.py

@@ -88,7 +88,7 @@ class ILSSurrogate(Algorithm):
         print("Start fitting again the surrogate model")
         for r in range(10):
             print("Iteration n°{0}: for fitting surrogate".format(r))
-            algo.run(samplefile=self.solutions_file, sample=1000, step=10)
+            algo.run(samplefile=self.solutions_file, sample=100, step=10)
 
         joblib.dump(algo, self.surrogate_file_path)