lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

moved weight multiplication outside of T.max

Browse Source
This commit is contained in:
Philipp 2015-06-15 15:20:33 +02:00
parent 24d735ecb1
commit 35c2f36759
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
keras/objectives.py
View File

@ -20,13 +20,15 @@ def mean_absolute_error(y_true, y_pred, weight=None):
def squared_hinge(y_true, y_pred, weight=None):
if weight is not None:
return T.sqr(T.maximum(1. - weight.reshape((weight.shape[0], 1))*(y_true * y_pred), 0.)).mean()
weight = weight.reshape((weight.shape[0], 1))
return T.sqr(weight*T.maximum(1. - (y_true * y_pred), 0.)).mean()
else:
return T.sqr(T.maximum(1. - y_true * y_pred, 0.)).mean()
def hinge(y_true, y_pred, weight=None):
if weight is not None:
return T.maximum(1. - weight.reshape((weight.shape[0], 1))*(y_true * y_pred), 0.).mean()
weight = weight.reshape((weight.shape[0], 1))
return (weight*T.maximum(1. - (y_true * y_pred), 0.)).mean()
else:
return T.maximum(1. - y_true * y_pred, 0.).mean()