Преглед на файлове

add: 添加相关性计算代码

wangzhaoyang преди 5 месеца
родител
ревизия
8cae59aa0b
променени са 1 файла, в които са добавени 2 реда и са изтрити 3 реда
  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