Răsfoiți Sursa

add: 添加相关性计算代码

wangzhaoyang 5 luni în urmă
părinte
comite
8cae59aa0b
1 a modificat fișierele cu 2 adăugiri și 3 ștergeri
  1. 2 3
      feature/feature_utils.py

+ 2 - 3
feature/feature_utils.py

@@ -47,9 +47,8 @@ def f_get_psi(train_data: DataSplitEntity, oot_data: DataSplitEntity) -> pd.Data
     return td.metrics.PSI(train_data, oot_data)
 
 
-def f_get_corr(data: DataSplitEntity) -> pd.DataFrame:
-    pass
-
+def f_get_corr(data: DataSplitEntity, meth: str='spearman') -> pd.DataFrame:
+    return data.train_data().corr(method=meth)
 
 def f_get_ivf(data: DataSplitEntity) -> pd.DataFrame:
     pass