This commit is contained in:
art 2019-10-02 20:56:06 -05:00
parent 4d01067a57
commit 484adfcda8

View File

@ -295,6 +295,11 @@ def regression_engine(device, inputs, outputs, args, loss = torch.nn.MSELoss(),
return regressions
#@jit TODO: determine jit type
def elo(starting_score):
@jit(forceobj=True)
def r_squared(predictions, targets): # assumes equal size inputs