Parcourir la source

update number of sample used for surrogate

Jérôme BUISINE il y a 3 ans
Parent
commit
2144b1d5da
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  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)