소스 검색

Merge branch 'dev_david' of anweiguo/algorithm_platform into dev_awg

liudawei 1 개월 전
부모
커밋
eeb0351120
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      evaluation_processing/analysis_report.py

+ 1 - 1
evaluation_processing/analysis_report.py

@@ -97,7 +97,7 @@ def put_analysis_report_to_html(args, df_clean, df_predict, df_accuracy):
     # -------------------- 生成相关性热力图 --------------------
 
     # 计算相关矩阵
-    correlation_matrix = df_clean_after.corr()
+    correlation_matrix = df_clean_after.select_dtypes(include=['number']).corr()
 
     # 生成热力图,带数值标签和新配色
     fig_heatmap = go.Figure(data=go.Heatmap(