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

introduce party color mapping to make colors more consistent

Browse Source
This commit is contained in:
Hannes Kuchelmeister
2021-06-12 11:30:28 +02:00
parent f48c5839db
commit 215acd1538
4 changed files with 41 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
de_analysis.py
View File

@@ -4,6 +4,7 @@
import voting_lib.load_data as ld
import voting_lib.voting_analysis as va
import voting_lib.political_compass as pc
from voting_lib.party_colors import de_name_color
import numpy as np
# Training Paramters
@@ -30,5 +31,5 @@ for period, df in dataset.items():
model = va.train_model(X, grid_h, grid_w, radius, step, ep)
# Predict and visualize output
va.predict(model, data, grid_h, grid_w, pc.get_compass_parties(year=period_to_compass_year[period], country='de'))
va.predict(model, data, grid_h, grid_w, de_name_color, pc.get_compass_parties(year=period_to_compass_year[period], country='de'))