Jelajahi Sumber

add: 添加toad的IV计算代码

wangzhaoyang 5 bulan lalu
induk
melakukan
5d06b2b085
1 mengubah file dengan 2 tambahan dan 2 penghapusan
  1. 2 2
      feature/feature_utils.py

+ 2 - 2
feature/feature_utils.py

@@ -39,8 +39,8 @@ def f_get_woe(train_selected: DataSplitEntity, test_selected: DataSplitEntity, o
     return train_woe, test_woe, oot_woe
 
 def f_get_iv(data: DataSplitEntity) -> pd.DataFrame:
-    pass
-
+    # 计算前,先排除掉不需要计算IV的cols
+    return td.quality(data, 'target',iv_only=True)
 
 def f_get_psi(train_data: DataSplitEntity, oot_data: DataSplitEntity) -> pd.DataFrame:
     # 计算前,先排除掉不需要的cols