sax.nn.loss module

Contents

sax.nn.loss module#

SAX Loss Functions

mse(x, y)[source]#

mean squared error

Parameters:

x (ComplexArrayND) –
y (ComplexArrayND) –

Return type:

float

huber_loss(x, y, delta=0.5)[source]#

huber loss

Parameters:

x (ComplexArrayND) –
y (ComplexArrayND) –
delta (float) –

Return type:

float

l2_reg(weights)[source]#

L2 regularization loss

Parameters:: weights (Dict[str, ComplexArrayND]) –
Return type:: float