German: Same party names over periods

This commit is contained in:
Deepthi Pathare
2021-05-24 19:10:31 +02:00
parent d1033f9443
commit 653e773f39
2 changed files with 5 additions and 2 deletions

View File

@@ -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()

View File

@@ -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]