mirror of
https://github.com/titanscouting/tra-analysis.git
synced 2024-12-27 01:59:08 +00:00
comments
This commit is contained in:
parent
eade12bb71
commit
61e2cad206
@ -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 = []
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user