mirror of
https://github.com/titanscouting/tra-analysis.git
synced 2024-12-27 01:59:08 +00:00
something changed idk what
This commit is contained in:
parent
4e50a79614
commit
32bc329e91
Binary file not shown.
@ -317,11 +317,6 @@ def histo_analysis(hist_data):
|
|||||||
@jit(forceobj=True)
|
@jit(forceobj=True)
|
||||||
def regression(device, inputs, outputs, args, loss = torch.nn.MSELoss(), _iterations = 10000, lr = 0.01, _iterations_ply = 10000, lr_ply = 0.01, power_limit = None): # inputs, outputs expects N-D array
|
def regression(device, inputs, outputs, args, loss = torch.nn.MSELoss(), _iterations = 10000, lr = 0.01, _iterations_ply = 10000, lr_ply = 0.01, power_limit = None): # inputs, outputs expects N-D array
|
||||||
|
|
||||||
if power_limit == None:
|
|
||||||
power_limit = len(outputs)
|
|
||||||
else:
|
|
||||||
power_limit += 1
|
|
||||||
|
|
||||||
regressions = []
|
regressions = []
|
||||||
Regression().set_device(device)
|
Regression().set_device(device)
|
||||||
|
|
||||||
@ -346,7 +341,7 @@ def regression(device, inputs, outputs, args, loss = torch.nn.MSELoss(), _iterat
|
|||||||
|
|
||||||
if power_limit == None:
|
if power_limit == None:
|
||||||
|
|
||||||
power_limit = len(outputs[0])
|
power_limit = len(outputs[0]) - 1
|
||||||
|
|
||||||
for i in range(2, power_limit):
|
for i in range(2, power_limit):
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user