modify: 修改分箱代码

feature/feature_utils.py

@@ -19,8 +19,8 @@ def f_get_bins(data: DataSplitEntity, feat: str, strategy: str='quantile', nbins
         return feature_binned.astype(int).astype(str)
     # 等宽分箱
     if strategy == BinsStrategyEnum.WIDTH.value:
-        bin_width = (data[feat].max() - data[feat].min()) / nbins
-        return pd.cut(data[feat], bins=nbins, labels=[f'Bin_{i}' for i in range(1, nbins + 1)])
+        bin_width = (data.train_data()[feat].max() - data.train_data()[feat].min()) / nbins
+        return pd.cut(data.train_data()[feat], bins=nbins, labels=[f'Bin_{i}' for i in range(1, nbins + 1)])
     # 使用toad分箱
     '''
     c = td.transfrom.Combiner()