Browse Source

add: 添加分箱代码

wangzhaoyang 5 tháng trước cách đây
mục cha
commit
eaa02f9169
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  1. 1 1
      feature/feature_utils.py

+ 1 - 1
feature/feature_utils.py

@@ -9,7 +9,7 @@ from sklearn.preprocessing import KBinsDiscretizer
 from entitys import DataSplitEntity
 
 
-def f_get_bins(data: DataSplitEntity) -> pd.DataFrame:
+def f_get_bins(data: DataSplitEntity, feat: str, strategy: str='quantile', nbins: int=10) -> pd.DataFrame:
     # 等频分箱
     if strategy == 'quantile':
         kbin_encoder = KBinsDiscretizer(n_bins=nbins, encode='ordinal', strategy='quantile')