Pārlūkot izejas kodu

use of tail too for learn dataset

Jérôme BUISINE 3 gadi atpakaļ
vecāks
revīzija
1baf7bbbb0
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. 1 1
      optimization/ILSPopSurrogate.py

+ 1 - 1
optimization/ILSPopSurrogate.py

@@ -134,7 +134,7 @@ class ILSPopSurrogate(Algorithm):
         reduced_df = shuffle(reduced_df)
 
         # shuffle dataset
-        learn = reduced_df.iloc[0:ntraining_samples]
+        learn = reduced_df.tail(ntraining_samples)
         test = reduced_df.drop(learn.index)
 
         print("Start fitting again the surrogate model")