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

add party affiliation to uk data

Browse Source
This commit is contained in:
Hannes Kuchelmeister
2021-05-04 11:47:46 +02:00
parent 2144d5913f
commit fe8b2ba0e7
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
VotingAnalysis.py
View File

@@ -55,6 +55,7 @@ def load_uk_data():
data = pd.DataFrame() data = pd.DataFrame()
name_column = 'Member' name_column = 'Member'
party_column = 'Party'
vote_column = 'Vote' vote_column = 'Vote'
column_to_filename = {} column_to_filename = {}
@@ -83,7 +84,7 @@ def load_uk_data():
if data.empty: if data.empty:
# if first file that is loaded set data equal to data from first file # if first file that is loaded set data equal to data from first file
data = df[[name_column, vote_column_name]] data = df[[name_column, party_column, vote_column_name]]
else: else:
# merge data with already loaded data # merge data with already loaded data
data = data.merge(df[[name_column, vote_column_name]], on=name_column) data = data.merge(df[[name_column, vote_column_name]], on=name_column)
@@ -262,7 +263,7 @@ plt.style.use('ggplot')
# Load data # Load data
data = load_uk_data().to_numpy() data = load_uk_data().to_numpy()
X = data[:,1:] X = data[:,2:]
print(X) print(X)
inp = X.shape[1] # No of features (bill count) inp = X.shape[1] # No of features (bill count)