Ver Fonte

add: 添加toad的PSI计算代码

wangzhaoyang há 5 meses atrás
pai
commit
5fae047a0a
1 ficheiros alterados com 3 adições e 2 exclusões
  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: