瀏覽代碼

update drop learn set

Jérôme BUISINE 3 年之前
父節點
當前提交
2bcbe3880e
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      optimization/ILSPopSurrogate.py

+ 1 - 1
optimization/ILSPopSurrogate.py

@@ -135,7 +135,7 @@ class ILSPopSurrogate(Algorithm):
 
         # shuffle dataset
         learn = reduced_df.iloc[0:ntraining_samples]
-        test = reduced_df.drop[learn.index]
+        test = reduced_df.drop(learn.index)
 
         print("Start fitting again the surrogate model")
         print(f'Using {ntraining_samples} samples of {max_samples} for train dataset')