Explorar el Código

add: 添加分箱代码

wangzhaoyang hace 5 meses
padre
commit
eaa02f9169
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  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')