From 653e773f39016834f40ba1c3379a7bc79407a2ce Mon Sep 17 00:00:00 2001 From: Deepthi Pathare Date: Mon, 24 May 2021 19:10:31 +0200 Subject: [PATCH] German: Same party names over periods --- de_analysis.py | 4 ++-- voting_lib/load_data.py | 3 +++ 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/de_analysis.py b/de_analysis.py index f15909d..456e165 100755 --- a/de_analysis.py +++ b/de_analysis.py @@ -11,8 +11,8 @@ import numpy as np grid_h = 11 # Grid height grid_w = 11 # Grid width radius = 2 # Neighbour radius -step = 0.5 -ep = 100 # No of epochs +step = 0.5 # Learning step +ep = 300 # No of epochs # Load data dataset = ld.load_german_data() diff --git a/voting_lib/load_data.py b/voting_lib/load_data.py index dbc7807..2ec0106 100755 --- a/voting_lib/load_data.py +++ b/voting_lib/load_data.py @@ -20,6 +20,8 @@ def load_german_data(): party_column_g = 'Fraktion/Gruppe' name_column = 'Member' party_column = 'Party' + + parties_to_replace = {'BÜNDNIS`90/DIE GRÜNEN':'BÜ90/GR'} parties_to_remove = ['Fraktionslos'] @@ -48,6 +50,7 @@ def load_german_data(): df=df.rename(columns={name_column_g:name_column,party_column_g:party_column}) df=df[~df.Party.isin(parties_to_remove)] + df[party_column].replace(parties_to_replace, inplace=True) period = df.iloc[0][period_column_g]