mirror of
https://github.com/titanscouting/tra-analysis.git
synced 2026-10-12 05:08:31 +00:00
comments
This commit is contained in:
1 parent
3a17ac5154
commit
af20fb0fa7
1 file changed
+1
-1
@@ -248,7 +248,7 @@ def histo_analysis(hist_data):
|
||||
return basic_stats(derivative)[0], basic_stats(derivative)[3]
|
||||
|
||||
@jit(forceobj=True)
|
||||
def regression(device, inputs, outputs, args, loss = torch.nn.MSELoss(), _iterations = 10000, lr = 0.01):
|
||||
def regression(device, inputs, outputs, args, loss = torch.nn.MSELoss(), _iterations = 10000, lr = 0.01): # inputs, outputs expects N-D array
|
||||
|
||||
regressions = []
|
||||
|
||||
|
||||
Reference in new issue
Block a user