소스 검색

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