From 7fd22ccbc63c7f1563f1b508467c6093070fa004 Mon Sep 17 00:00:00 2001 From: Hannes Kuchelmeister Date: Thu, 20 May 2021 13:36:13 +0200 Subject: [PATCH] load political compass data for uk in uk analysis --- uk_analysis.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/uk_analysis.py b/uk_analysis.py index f7955bd..807e7db 100644 --- a/uk_analysis.py +++ b/uk_analysis.py @@ -5,6 +5,7 @@ import voting_lib.load_data as ld import voting_lib.voting_analysis as va +import voting_lib.political_compass as pc import numpy as np import pandas as pd import os @@ -16,14 +17,15 @@ grid_h = 30 # Grid height grid_w = 30 # Grid width radius = 3 # Neighbour radius step = 0.5 -ep = 100 # No of epochs +ep = 1 # No of epochs +period_to_compass_year = {'2015_uk':2015, '2017_uk':2017, '2019_uk':2019} main_directory = 'uk/csv' for dirname, _, filenames in os.walk(main_directory): if dirname == main_directory: #to skip main directory path continue - else: + elif os.path.isdir(dirname): # Load data data = ld.load_uk_data(dirname).to_numpy() @@ -31,4 +33,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) \ No newline at end of file + va.predict(model, data, grid_h, grid_w, pc.get_compass_parties(year=period_to_compass_year[dirname.split('/')[-1]], country='uk')) \ No newline at end of file