Explorar o código

bugfix: 粗分箱时切分点越界

yq hai 2 meses
pai
achega
4567173b70
Modificáronse 1 ficheiros con 2 adicións e 1 borrados
  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)