update

pages/2_Evaluation.py

@@ -82,8 +82,9 @@ def app():
                         x='variable', y='value', color='variable', title='Spread of Ranks')
                     st.plotly_chart(box_rank_fig)
 
-                    corr_fig = create_correlation_heatmaps(df)
-                    st.plotly_chart(corr_fig)
+                    heatmaps = create_correlation_heatmaps(df)
+                    for title, fig in heatmaps.items():
+                        st.plotly_chart(fig)
 
                     st.download_button(
                         label="Download Evaluation Results",

util/plot.py

@@ -84,67 +84,19 @@ def create_correlation_heatmaps(df):
     scores_corr_kendall = scores_df.corr(method='kendall')
     ranks_corr_kendall = ranks_df.corr(method='kendall')
 
-    # Plotting the heatmaps
-    fig = go.Figure()
-
-    # fig.add_trace(go.Heatmap(
-    #     z=scores_corr_pearson.values,
-    #     x=scores_corr_pearson.columns,
-    #     y=scores_corr_pearson.index,
-    #     colorscale='Viridis',
-    #     showscale=True,
-    #     name='Scores Pearson Correlation'
-    # ))
-
-    fig.add_trace(go.Heatmap(
-        z=ranks_corr_pearson.values,
-        x=ranks_corr_pearson.columns,
-        y=ranks_corr_pearson.index,
-        colorscale='Viridis',
-        showscale=True,
-        name='Ranks Pearson Correlation'
-    ))
-
-    # fig.add_trace(go.Heatmap(
-    #     z=scores_corr_spearman.values,
-    #     x=scores_corr_spearman.columns,
-    #     y=scores_corr_spearman.index,
-    #     colorscale='Cividis',
-    #     showscale=True,
-    #     name='Scores Spearman Correlation'
-    # ))
-    #
-    # fig.add_trace(go.Heatmap(
-    #     z=ranks_corr_spearman.values,
-    #     x=ranks_corr_spearman.columns,
-    #     y=ranks_corr_spearman.index,
-    #     colorscale='Cividis',
-    #     showscale=True,
-    #     name='Ranks Spearman Correlation'
-    # ))
-
-    # fig.add_trace(go.Heatmap(
-    #     z=scores_corr_kendall.values,
-    #     x=scores_corr_kendall.columns,
-    #     y=scores_corr_kendall.index,
-    #     colorscale='Inferno',
-    #     showscale=True,
-    #     name='Scores Kendall Correlation'
-    # ))
-    #
-    # fig.add_trace(go.Heatmap(
-    #     z=ranks_corr_kendall.values,
-    #     x=ranks_corr_kendall.columns,
-    #     y=ranks_corr_kendall.index,
-    #     colorscale='Inferno',
-    #     showscale=True,
-    #     name='Ranks Kendall Correlation'
-    # ))
-
-    # Update layout
-    fig.update_layout(
-        title='Correlation Heatmaps',
-        xaxis_nticks=36
-    )
-
-    return fig
+    # Plotting the heatmaps separately
+    heatmaps = {
+        'Scores Pearson Correlation': scores_corr_pearson,
+        'Ranks Pearson Correlation': ranks_corr_pearson,
+        'Scores Spearman Correlation': scores_corr_spearman,
+        'Ranks Spearman Correlation': ranks_corr_spearman,
+        'Scores Kendall Correlation': scores_corr_kendall,
+        'Ranks Kendall Correlation': ranks_corr_kendall
+    }
+
+    figs = {}
+    for title, corr_matrix in heatmaps.items():
+        fig = px.imshow(corr_matrix, text_auto=True, title=title)
+        figs[title] = fig
+
+    return figs