Ver Fonte

add: 添加分箱代码

wangzhaoyang há 5 meses atrás
pai
commit
eaa02f9169
1 ficheiros alterados com 1 adições e 1 exclusões
  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')