anweiguo 1 ヶ月 前
コミット
dc7c9434a3
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(