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

Correction in calculation of parties position

Browse Source
This commit is contained in:
Deepthi Pathare
2021-05-11 17:36:29 +02:00
parent 77004cd32d
commit 0f74c92fa0
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
voting_lib/voting_analysis.py
View File

@@ -39,12 +39,11 @@ def predict(model, data, grid_h, grid_w):
# plotting mps
party_affiliation = data[:,1]
plot_mps(data[:,0], xs, ys, party_affiliation)
xs_disp, ys_disp = plot_mps(data[:,0], xs, ys, party_affiliation)
plt.show()
# calculating party positions based on mps
party_pos = calc_party_pos(np.column_stack((xs, ys)), party_affiliation)
party_pos = calc_party_pos(np.column_stack((xs_disp, ys_disp)), party_affiliation)
# plotting parties
plot_parties(party_pos)
@@ -155,11 +154,11 @@ def plot_mps(names, xs, ys, party_affiliation):
party_index_mapping, party_ids = np.unique(party_affiliation, return_inverse=True)
# add random offset to show points that are in the same location
ys_disp = ys + np.random.rand(ys.shape[0])
xs_disp = xs + np.random.rand(xs.shape[0])
ys_disp = ys + np.random.rand(ys.shape[0])
parties = party_index_mapping[party_ids]
plot_hoverscatter(xs_disp, ys_disp, names + " (" + parties + ")", party_ids)
return xs_disp, ys_disp
def calc_party_pos(members_of_parliament, party_affiliation):
party_index_mapping, party_ids = np.unique(party_affiliation, return_inverse=True)