瀏覽代碼

bugfix: 粗分箱时切分点越界

yq 2 月之前
父節點
當前提交
4567173b70
共有 1 個文件被更改,包括 2 次插入1 次删除
  1. 2 1
      feature/strategy_iv.py

+ 2 - 1
feature/strategy_iv.py

@@ -241,6 +241,7 @@ class StrategyIv(FilterStrategyBase):
         train_data_filter = train_data_filter.sort_values(by=x_column, ascending=True)
         train_data_x = train_data_filter[x_column]
         train_data_x_describe = train_data_x.describe(percentiles=[0.1, 0.9])
+        train_data_x_max = train_data_x.max()
 
         test_data = data.test_data
         test_data_filter = None
@@ -275,7 +276,7 @@ class StrategyIv(FilterStrategyBase):
                 point = round(point, 2)
                 if point == 0:
                     continue
-                if point not in point_list_cache:
+                if point not in point_list_cache and point < train_data_x_max:
                     point_list_cache.append(point)
             if point_list_cache not in points_list and len(point_list_cache) != 0:
                 points_list.append(point_list_cache)