Zekun Wu commited on
Commit
6bf4517
1 Parent(s): bfdd3c9
Files changed (1) hide show
  1. util/plot.py +43 -43
util/plot.py CHANGED
@@ -87,14 +87,14 @@ def create_correlation_heatmaps(df):
87
  # Plotting the heatmaps
88
  fig = go.Figure()
89
 
90
- fig.add_trace(go.Heatmap(
91
- z=scores_corr_pearson.values,
92
- x=scores_corr_pearson.columns,
93
- y=scores_corr_pearson.index,
94
- colorscale='Viridis',
95
- showscale=True,
96
- name='Scores Pearson Correlation'
97
- ))
98
 
99
  fig.add_trace(go.Heatmap(
100
  z=ranks_corr_pearson.values,
@@ -105,41 +105,41 @@ def create_correlation_heatmaps(df):
105
  name='Ranks Pearson Correlation'
106
  ))
107
 
108
- fig.add_trace(go.Heatmap(
109
- z=scores_corr_spearman.values,
110
- x=scores_corr_spearman.columns,
111
- y=scores_corr_spearman.index,
112
- colorscale='Cividis',
113
- showscale=True,
114
- name='Scores Spearman Correlation'
115
- ))
116
-
117
- fig.add_trace(go.Heatmap(
118
- z=ranks_corr_spearman.values,
119
- x=ranks_corr_spearman.columns,
120
- y=ranks_corr_spearman.index,
121
- colorscale='Cividis',
122
- showscale=True,
123
- name='Ranks Spearman Correlation'
124
- ))
125
-
126
- fig.add_trace(go.Heatmap(
127
- z=scores_corr_kendall.values,
128
- x=scores_corr_kendall.columns,
129
- y=scores_corr_kendall.index,
130
- colorscale='Inferno',
131
- showscale=True,
132
- name='Scores Kendall Correlation'
133
- ))
134
-
135
- fig.add_trace(go.Heatmap(
136
- z=ranks_corr_kendall.values,
137
- x=ranks_corr_kendall.columns,
138
- y=ranks_corr_kendall.index,
139
- colorscale='Inferno',
140
- showscale=True,
141
- name='Ranks Kendall Correlation'
142
- ))
143
 
144
  # Update layout
145
  fig.update_layout(
 
87
  # Plotting the heatmaps
88
  fig = go.Figure()
89
 
90
+ # fig.add_trace(go.Heatmap(
91
+ # z=scores_corr_pearson.values,
92
+ # x=scores_corr_pearson.columns,
93
+ # y=scores_corr_pearson.index,
94
+ # colorscale='Viridis',
95
+ # showscale=True,
96
+ # name='Scores Pearson Correlation'
97
+ # ))
98
 
99
  fig.add_trace(go.Heatmap(
100
  z=ranks_corr_pearson.values,
 
105
  name='Ranks Pearson Correlation'
106
  ))
107
 
108
+ # fig.add_trace(go.Heatmap(
109
+ # z=scores_corr_spearman.values,
110
+ # x=scores_corr_spearman.columns,
111
+ # y=scores_corr_spearman.index,
112
+ # colorscale='Cividis',
113
+ # showscale=True,
114
+ # name='Scores Spearman Correlation'
115
+ # ))
116
+ #
117
+ # fig.add_trace(go.Heatmap(
118
+ # z=ranks_corr_spearman.values,
119
+ # x=ranks_corr_spearman.columns,
120
+ # y=ranks_corr_spearman.index,
121
+ # colorscale='Cividis',
122
+ # showscale=True,
123
+ # name='Ranks Spearman Correlation'
124
+ # ))
125
+
126
+ # fig.add_trace(go.Heatmap(
127
+ # z=scores_corr_kendall.values,
128
+ # x=scores_corr_kendall.columns,
129
+ # y=scores_corr_kendall.index,
130
+ # colorscale='Inferno',
131
+ # showscale=True,
132
+ # name='Scores Kendall Correlation'
133
+ # ))
134
+ #
135
+ # fig.add_trace(go.Heatmap(
136
+ # z=ranks_corr_kendall.values,
137
+ # x=ranks_corr_kendall.columns,
138
+ # y=ranks_corr_kendall.index,
139
+ # colorscale='Inferno',
140
+ # showscale=True,
141
+ # name='Ranks Kendall Correlation'
142
+ # ))
143
 
144
  # Update layout
145
  fig.update_layout(