浏览代码

fix float index..

Jérôme BUISINE 3 年之前
父节点
当前提交
3a61612fee
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      optimization/ILSPopSurrogate.py

+ 1 - 1
optimization/ILSPopSurrogate.py

@@ -127,7 +127,7 @@ class ILSPopSurrogate(Algorithm):
         if df.x.count() < max_samples:
             max_samples = df.x.count()
 
-        ntraining_samples = max_samples * 0.80
+        ntraining_samples = int(max_samples * 0.80)
         
         # extract reduced dataset if necessary
         reduced_df = df.tail(max_samples)