alu/tra-analysis
1
0
Fork 0
mirror of https://github.com/titanscouting/tra-analysis.git synced 2024-12-27 01:59:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix PolyRegKernel

Browse Source
This commit is contained in:
jlevine18 2019-10-09 22:23:56 -05:00 committed by GitHub
parent 8eac3d5af1
commit 7c121d48fc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
data analysis/analysis/analysis.py
View File

@ -533,10 +533,13 @@ class Regression:
def take_all_pwrs(self, vec, pwr): def take_all_pwrs(self, vec, pwr):
#todo: vectorize (kinda) #todo: vectorize (kinda)
combins=torch.combinations(vec, r=pwr, with_replacement=True) combins=torch.combinations(vec, r=pwr, with_replacement=True)
out=torch.ones(combins.size()[0]) out=torch.ones(combins.size()[0]).to(device).to(torch.float)
for i in torch.t(combins): for i in torch.t(combins).to(device).to(torch.float):
out *= i out *= i
return torch.cat(out,take_all_pwrs(vec, pwr-1)) if pwr == 1:
return out
else:
return torch.cat((out,self.take_all_pwrs(vec, pwr-1)))
def forward(self,mtx): def forward(self,mtx):
#TODO: Vectorize the last part #TODO: Vectorize the last part
cols=[] cols=[]