alu/tra-analysis
1
0
Fork 0
mirror of https://github.com/titanscouting/tra-analysis.git synced 2024-09-20 08:57:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

WORKING!!!!

Browse Source
This commit is contained in:
ltcptgeneral 2019-03-21 17:25:16 -05:00
parent 4cc1ad28fe
commit 52c3ddd2d6
2 changed files with 35 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
data analysis/__pycache__/analysis.cpython-37.pyc
View File

Binary file not shown.

64
data analysis/superscript.py
View File

@ -67,11 +67,13 @@ def titanservice():
file_list = glob.glob(source_dir + '/*.csv') #supposedly sorts by alphabetical order, skips reading teams.csv because of redundancy
data = []
files = [fn for fn in glob.glob('data/*.csv')
if not (os.path.basename(fn).startswith('teams') or os.path.basename(fn).startswith('match') or os.path.basename(fn).startswith('notes') or os.path.basename(fn).startswith('observationType') or os.path.basename(fn).startswith('teamDBRef'))] #scores will be handled sperately
if not (os.path.basename(fn).startswith('scores') or os.path.basename(fn).startswith('teams') or os.path.basename(fn).startswith('match') or os.path.basename(fn).startswith('notes') or os.path.basename(fn).startswith('observationType') or os.path.basename(fn).startswith('teamDBRef'))] #scores will be handled sperately
for i in files:
data.append(analysis.load_csv(i))
#print(files)
stats = []
measure_stats = []
teams = analysis.load_csv("data/teams.csv")
@ -127,49 +129,53 @@ def titanservice():
stats.append(list(measure_stats))
nishant = []
for i in range(len(scores)):
for i in range(len(scores)):
ofbest_curve = [None]
r2best_curve = [None]
#print(scores)
line = measure[i]
ofbest_curve = [None]
r2best_curve = [None]
#print(line)
line = scores[i]
x = list(range(len(line)))
eqs, rmss, r2s, overfit = analysis.optimize_regression(x, line, 10, 1)
#print(line)
beqs, brmss, br2s, boverfit = analysis.select_best_regression(eqs, rmss, r2s, overfit, "min_overfit")
#print(line)
#print(eqs, rmss, r2s, overfit)
x = list(range(len(line)))
eqs, rmss, r2s, overfit = analysis.optimize_regression(x, line, 10, 1)
ofbest_curve.append(beqs)
ofbest_curve.append(brmss)
ofbest_curve.append(br2s)
ofbest_curve.append(boverfit)
ofbest_curve.pop(0)
beqs, brmss, br2s, boverfit = analysis.select_best_regression(eqs, rmss, r2s, overfit, "min_overfit")
#print(ofbest_curve)
#print(eqs, rmss, r2s, overfit)
beqs, brmss, br2s, boverfit = analysis.select_best_regression(eqs, rmss, r2s, overfit, "max_r2s")
ofbest_curve.append(beqs)
ofbest_curve.append(brmss)
ofbest_curve.append(br2s)
ofbest_curve.append(boverfit)
ofbest_curve.pop(0)
r2best_curve.append(beqs)
r2best_curve.append(brmss)
r2best_curve.append(br2s)
r2best_curve.append(boverfit)
r2best_curve.pop(0)
#print(ofbest_curve)
#print(r2best_curve)
beqs, brmss, br2s, boverfit = analysis.select_best_regression(eqs, rmss, r2s, overfit, "max_r2s")
z = len(scores[0]) + 1
nis_num = []
r2best_curve.append(beqs)
r2best_curve.append(brmss)
r2best_curve.append(br2s)
r2best_curve.append(boverfit)
r2best_curve.pop(0)
nis_num.append(eval(str(ofbest_curve[0])))
nis_num.append(eval(str(r2best_curve[0])))
#print(r2best_curve)
nis_num.append((eval(ofbest_curve[0]) + eval(r2best_curve[0])) / 2)
z = len(scores[0]) + 1
nis_num = []
nishant.append(teams[i] + nis_num)
nis_num.append(eval(str(ofbest_curve[0])))
nis_num.append(eval(str(r2best_curve[0])))
nis_num.append((eval(ofbest_curve[0]) + eval(r2best_curve[0])) / 2)
nishant.append(teams[i] + nis_num)
json_out = {}
score_out = {}