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

Data split based on election period and model is trained on each set, closes #15

Browse Source
This commit is contained in:
Deepthi Pathare
2021-05-11 23:59:08 +02:00
parent 69759ccd3b
commit 82cef720fc
3 changed files with 42 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
voting_lib/voting_analysis.py
View File

@@ -60,14 +60,14 @@ def predict(model, data, grid_h, grid_w):
plot_party_distances(part_distance_in)
plt.show()
# Heatmap of weights
plt.figure()
weight = model.weight.reshape((model.n_inputs, grid_h, grid_w))
heatmap = compute_heatmap(weight, grid_h, grid_w)
plt.imshow(heatmap, cmap='Greys_r', interpolation='nearest')
plt.axis('off')
plt.colorbar()
plt.show()
## Heatmap of weights
# plt.figure()
# weight = model.weight.reshape((model.n_inputs, grid_h, grid_w))
# heatmap = compute_heatmap(weight, grid_h, grid_w)
# plt.imshow(heatmap, cmap='Greys_r', interpolation='nearest')
# plt.axis('off')
# plt.colorbar()
# plt.show()
def iter_neighbours(weights, hexagon=False):
_, grid_height, grid_width = weights.shape