Updated UK analysis file

Executes multiple period datasets
This commit is contained in:
Tej Kedar
2021-05-18 20:57:45 +02:00
committed by GitHub
parent 1c2bfd7c41
commit 010815dca3

43
uk_analysis_new.py Normal file
View File

@@ -0,0 +1,43 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#This code is modified to run in Kaggle
#import voting_lib.load_data as ld
#import voting_lib.voting_analysis as va
import numpy as np
import pandas as pd
import os
# Train model
grid_h = 30 # Grid height
grid_w = 30 # Grid width
radius = 3 # Neighbour radius
step = 0.5
ep = 100 # No of epochs
#main directory path(should contain differnt dataset directory) can be changed
main_directory = '/kaggle/input'
for dirname, _, filenames in os.walk(main_directory):
#print(os.path.join(dirname))
if dirname == main_directory: #to skip main directory path
continue
else:
# Load data
#data = ld.load_uk_data().to_numpy()
#modifiy load_data.py --> load_uk_data() to load_uk_data(path)
# --> Place path in directory -> for dirname, _, filenames in os.walk(path):
data = load_uk_data(dirname).to_numpy()
X = data[:,2:]
#model = va.train_model(X, grid_h, grid_w, radius, step, ep)
model = train_model(X, grid_h, grid_w, radius, step, ep)
# Predict and visualize output
#va.predict(model, data, grid_h, grid_w)
predict(model, data, grid_h, grid_w)