alu/tra-analysis
1
0 Fork 0
mirror of https://github.com/titanscouting/tra-analysis.git synced 2026-10-12 05:08:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

quick change

Browse Source
This commit is contained in:
art committed 2019-10-04 10:37:29 -05:00
1 parent af20fb0fa7
commit a853e9b02b
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
data analysis/analysis/analysis.py
+1 -1
View File
@@ -256,7 +256,7 @@ def regression(device, inputs, outputs, args, loss = torch.nn.MSELoss(), _iterat
Regression.set_device(device) Regression.set_device(device)
if 'linear' in args: if 'lin' in args:
model = Regression.SGDTrain(Regression.LinearRegKernel(len(inputs)), torch.tensor(inputs).to(torch.float).cuda(), torch.tensor([outputs]).to(torch.float).cuda(), iterations=_iterations, learning_rate=lr, return_losses=True) model = Regression.SGDTrain(Regression.LinearRegKernel(len(inputs)), torch.tensor(inputs).to(torch.float).cuda(), torch.tensor([outputs]).to(torch.float).cuda(), iterations=_iterations, learning_rate=lr, return_losses=True)
regressions.append([model[0].parameters, model[1][::-1][0]]) regressions.append([model[0].parameters, model[1][::-1][0]])