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
uk_analysis.py
View File

@@ -6,6 +6,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 uk_name_color
import numpy as np
import pandas as pd
import os
@@ -33,4 +34,4 @@ for dirname, _, filenames in os.walk(main_directory):
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[dirname.split('/')[-1]], country='uk'))
va.predict(model, data, grid_h, grid_w, uk_name_color, pc.get_compass_parties(year=period_to_compass_year[dirname.split('/')[-1]], country='uk'))