Quellcode durchsuchen

Merge branch 'dev_david' of anweiguo/algorithm_platform into dev_awg

liudawei vor 1 Monat
Ursprung
Commit
eeb0351120
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  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(