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 data to dataset loading

Browse Source
This commit is contained in:
Hannes Kuchelmeister
2021-05-04 10:41:42 +02:00
parent b8c9f51063
commit 7ca3762e43
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

@@ -16,6 +16,7 @@ def load_german_data():
data = pd.DataFrame() data = pd.DataFrame()
name_column = 'Bezeichnung' name_column = 'Bezeichnung'
party_column = 'Fraktion/Gruppe'
vote_column_to_title = {} vote_column_to_title = {}
@@ -38,7 +39,7 @@ def load_german_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)
@@ -176,7 +177,7 @@ plt.style.use('ggplot')
# Load data # Load data
data = load_german_data().to_numpy() data = load_german_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)