Procházet zdrojové kódy

add: 添加toad的PSI计算代码

wangzhaoyang před 5 měsíci
rodič
revize
5fae047a0a
1 změnil soubory, kde provedl 3 přidání a 2 odebrání
  1. 3 2
      feature/feature_utils.py

+ 3 - 2
feature/feature_utils.py

@@ -42,8 +42,9 @@ def f_get_iv(data: DataSplitEntity) -> pd.DataFrame:
     pass
 
 
-def f_get_psi(data: DataSplitEntity) -> pd.DataFrame:
-    pass
+def f_get_psi(train_data: DataSplitEntity, oot_data: DataSplitEntity) -> pd.DataFrame:
+    # 计算前,先排除掉不需要的cols
+    return td.metrics.PSI(train_data, oot_data)
 
 
 def f_get_corr(data: DataSplitEntity) -> pd.DataFrame: