ソースを参照

add: 添加分箱代码

wangzhaoyang 5 ヶ月 前
コミット
eaa02f9169
1 ファイル変更1 行追加1 行削除
  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')