This commit is contained in:
art 2019-10-02 20:56:06 -05:00
parent f8267e17a5
commit 86c7f33f23

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