ソースを参照

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)