13hannes11/UU_NCML_Project
1
0
Fork 0
mirror of https://github.com/13hannes11/UU_NCML_Project.git synced 2024-09-03 20:50:59 +02:00
Code Issues Projects Releases Wiki Activity

adding alpha to heatmap to reduce contrast

Browse Source
This commit is contained in:
Hannes Kuchelmeister
2021-06-12 11:29:07 +02:00
parent 653e773f39
commit f48c5839db
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
voting_lib/voting_analysis.py
View File

@@ -59,7 +59,7 @@ def predict(model, data, grid_h, grid_w, comparison_data=pd.DataFrame()):
plt.figure() plt.figure()
weight = model.weight.reshape((model.n_inputs, grid_h, grid_w)) weight = model.weight.reshape((model.n_inputs, grid_h, grid_w))
heatmap = compute_heatmap(weight, grid_h, grid_w) heatmap = compute_heatmap(weight, grid_h, grid_w)
plt.imshow(heatmap, cmap ='Blues', interpolation='nearest',zorder=1) plt.imshow(heatmap, cmap ='Blues', interpolation='nearest',zorder=1, alpha=0.5)
plt.axis('off') plt.axis('off')
plt.colorbar() plt.colorbar()