mirror of
https://github.com/titanscouting/tra-analysis.git
synced 2024-09-20 08:57:56 +00:00
superscript.py - v 1.0.4.001
changelog: - grammar fixes
This commit is contained in:
ltcptgeneral
parent
a8f3598bbe
commit
57eda3815b
BIN
data analysis/__pycache__/superscript.cpython-37.pyc
Normal file
BIN
data analysis/__pycache__/superscript.cpython-37.pyc
Normal file
Binary file not shown.
@ -1,11 +1,13 @@
|
||||
#Titan Robotics Team 2022: Data Analysis Script
|
||||
#Titan Robotics Team 2022: Super Script
|
||||
#Written by Arthur Lu & Jacob Levine
|
||||
#Notes:
|
||||
#setup:
|
||||
|
||||
__version__ = "1.0.4.000"
|
||||
__version__ = "1.0.4.001"
|
||||
|
||||
__changelog__ = """changelog:
|
||||
1.0.4.001:
|
||||
- grammar fixes
|
||||
1.0.4.000:
|
||||
- actually pushes to firebase
|
||||
1.0.3.001:
|
||||
@ -37,82 +39,84 @@ import os
|
||||
import glob
|
||||
import numpy as np
|
||||
|
||||
# Use a service account
|
||||
cred = credentials.Certificate('keys/firebasekey.json')
|
||||
firebase_admin.initialize_app(cred)
|
||||
def titanservice():
|
||||
|
||||
db = firestore.client()
|
||||
# Use a service account
|
||||
cred = credentials.Certificate('keys/firebasekey.json')
|
||||
firebase_admin.initialize_app(cred)
|
||||
|
||||
#get all the data
|
||||
db = firestore.client()
|
||||
|
||||
analysis.generate_data("data/bdata.csv", 100, 5, -10, 10)
|
||||
#get all the data
|
||||
|
||||
source_dir = 'data'
|
||||
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')]
|
||||
analysis.generate_data("data/bdata.csv", 100, 5, -10, 10)
|
||||
|
||||
#for file_path in file_list:
|
||||
# if not os.path.basename(file_list).startswith("teams")
|
||||
# data.append(analysis.load_csv(file_path))
|
||||
source_dir = 'data'
|
||||
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')]
|
||||
|
||||
for i in files:
|
||||
data.append(analysis.load_csv(i))
|
||||
#for file_path in file_list:
|
||||
# if not os.path.basename(file_list).startswith("teams")
|
||||
# data.append(analysis.load_csv(file_path))
|
||||
|
||||
stats = []
|
||||
measure_stats = []
|
||||
teams = analysis.load_csv("data/teams.csv")
|
||||
for i in files:
|
||||
data.append(analysis.load_csv(i))
|
||||
|
||||
#assumes that team number is in the first column, and that the order of teams is the same across all files
|
||||
#unhelpful comment
|
||||
for measure in data: #unpacks 3d array into 2ds
|
||||
stats = []
|
||||
measure_stats = []
|
||||
teams = analysis.load_csv("data/teams.csv")
|
||||
|
||||
measure_stats = []
|
||||
#assumes that team number is in the first column, and that the order of teams is the same across all files
|
||||
#unhelpful comment
|
||||
for measure in data: #unpacks 3d array into 2ds
|
||||
|
||||
for i in range(len(measure)): #unpacks into specific teams
|
||||
measure_stats = []
|
||||
|
||||
ofbest_curve = [None]
|
||||
r2best_curve = [None]
|
||||
for i in range(len(measure)): #unpacks into specific teams
|
||||
|
||||
line = measure[i]
|
||||
ofbest_curve = [None]
|
||||
r2best_curve = [None]
|
||||
|
||||
#print(line)
|
||||
line = measure[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")
|
||||
x = list(range(len(line)))
|
||||
eqs, rmss, r2s, overfit = analysis.optimize_regression(x, line, 10, 1)
|
||||
|
||||
#print(eqs, rmss, r2s, overfit)
|
||||
beqs, brmss, br2s, boverfit = analysis.select_best_regression(eqs, rmss, r2s, overfit, "min_overfit")
|
||||
|
||||
ofbest_curve.append(beqs)
|
||||
ofbest_curve.append(brmss)
|
||||
ofbest_curve.append(br2s)
|
||||
ofbest_curve.append(boverfit)
|
||||
ofbest_curve.pop(0)
|
||||
#print(eqs, rmss, r2s, overfit)
|
||||
|
||||
#print(ofbest_curve)
|
||||
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, "max_r2s")
|
||||
#print(ofbest_curve)
|
||||
|
||||
r2best_curve.append(beqs)
|
||||
r2best_curve.append(brmss)
|
||||
r2best_curve.append(br2s)
|
||||
r2best_curve.append(boverfit)
|
||||
r2best_curve.pop(0)
|
||||
beqs, brmss, br2s, boverfit = analysis.select_best_regression(eqs, rmss, r2s, overfit, "max_r2s")
|
||||
|
||||
#print(r2best_curve)
|
||||
r2best_curve.append(beqs)
|
||||
r2best_curve.append(brmss)
|
||||
r2best_curve.append(br2s)
|
||||
r2best_curve.append(boverfit)
|
||||
r2best_curve.pop(0)
|
||||
|
||||
measure_stats.append(teams[i] + ["|"] + list(analysis.basic_stats(line, 0, 0)) + ["|"] + list(analysis.histo_analysis(line, 1, -3, 3)) + ["|"] + ofbest_curve + ["|"] + r2best_curve)
|
||||
#print(r2best_curve)
|
||||
|
||||
stats.append(list(measure_stats))
|
||||
measure_stats.append(teams[i] + ["|"] + list(analysis.basic_stats(line, 0, 0)) + ["|"] + list(analysis.histo_analysis(line, 1, -3, 3)) + ["|"] + ofbest_curve + ["|"] + r2best_curve)
|
||||
|
||||
json_out = {}
|
||||
stats.append(list(measure_stats))
|
||||
|
||||
for i in range(len(stats)):
|
||||
json_out[files[i]]=str(stats[i])
|
||||
json_out = {}
|
||||
|
||||
print(json_out)
|
||||
for i in range(len(stats)):
|
||||
json_out[files[i]]=str(stats[i])
|
||||
|
||||
db.collection(u'stats').document(u'stats-noNN').set(json_out)
|
||||
print(json_out)
|
||||
|
||||
db.collection(u'stats').document(u'stats-noNN').set(json_out)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user