alu/tra-superscript
1
0
Fork 0
mirror of https://github.com/titanscouting/tra-superscript.git synced 2024-11-10 06:54:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

finished rough outline for Match module

Browse Source 
Signed-off-by: Arthur Lu <learthurgo@gmail.com>

Former-commit-id: d5ebb0348b
This commit is contained in:
Arthur Lu 2021-10-21 21:27:35 +00:00
parent 79dd2c3479
commit 375b550153
1 changed files with 50 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
src/cli/module.py
View File

@ -1,4 +1,7 @@
import data as d import data as d
import signal
import numpy as np
import tra_analysis as an
class AutoVivification(dict): class AutoVivification(dict):
def __getitem__(self, item): def __getitem__(self, item):
@ -31,6 +34,7 @@ class Match:
teams = None teams = None
data = [] data = []
results = []
def __init__(self, config, apikey, tbakey, timestamp, teams): def __init__(self, config, apikey, tbakey, timestamp, teams):
self.config = config self.config = config
@ -63,10 +67,53 @@ class Match:
for competition in competitions: for competition in competitions:
for variable in self.config["tests"]: for variable in self.config["tests"]:
match_data = d.get_team_match_data(self.apikey, competition, team, variable) # needs modified implementation match_data = d.get_team_match_data(self.apikey, competition, team, variable) # needs modified implementation
self.data.append((team, competition, variable, match_data)) variable_tests = self.config["tests"][variable]
self.data.append({"team": team, "competition": competition, "variable": variable, "tests": variable_tests, "data": match_data})
def tests(test_data):
signal.signal(signal.SIGINT, signal.SIG_IGN)
if(test_data["data"] == None):
return None
data = np.array(test_data["data"])
data = data[np.isfinite(data)]
ranges = list(range(len(data)))
tests = test_data["tests"]
results = AutoVivification()
if "basic_stats" in tests:
results["basic_stats"] = an.basic_stats(data)
if "historical_analysis" in tests:
results["historical_analysis"] = an.histo_analysis([ranges, data])
if "regression_linear" in tests:
results["regression_linear"] = an.regression(ranges, data, ['lin'])
if "regression_logarithmic" in tests:
results["regression_logarithmic"] = an.regression(ranges, data, ['log'])
if "regression_exponential" in tests:
results["regression_exponential"] = an.regression(ranges, data, ['exp'])
if "regression_polynomial" in tests:
results["regression_polynomial"] = an.regression(ranges, data, ['ply'])
if "regression_sigmoidal" in tests:
results["regression_sigmoidal"] = an.regression(ranges, data, ['sig'])
return results
def process_data(self, exec_threads): def process_data(self, exec_threads):
pass self.results = list(exec_threads.map(self.tests, self.data))
def push_results(self): def push_results(self):
pass short_mapping = {"regression_linear": "lin", "regression_logarithmic": "log", "regression_exponential": "exp", "regression_polynomial": "ply", "regression_sigmoidal": "sig"}
results_short = AutoVivification()
i = 0
for result in self.results:
for variable in result:
if variable in short_mapping:
short = short_mapping[variable]
else:
short = variable
d.push_team_match_results(self.data[i]["team"], self.data[i]["competition"], self.data[i]["variable"], short, result[variable])
#results_short[ self.data["team"] ][ self.data["competition"] ][ self.data["variable"] ][short] = result[variable]
i+=1