|
@@ -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')
|