1
0

replaced np with K

This commit is contained in:
Kelvin Davis 2018-05-23 23:16:51 +10:00
parent 92eb1dce75
commit 70fe296360

View File

@ -53,16 +53,16 @@ def FCN():
return model
def precision(y_true, y_pred):
y_pred = np.round(y_pred)
num = np.sum(np.logical_and(y_true, y_pred))
den = np.sum(y_pred)
return np.divide(num, den)
y_pred = K.round(y_pred)
num = K.sum(K.logical_and(y_true, y_pred))
den = K.sum(y_pred)
return K.divide(num, den)
def recall(y_true, y_pred):
y_pred = np.round(y_pred)
num = np.sum(np.logical_and(y_true, y_pred))
den = np.sum(y_true)
return np.divide(num, den)
y_pred = K.round(y_pred)
num = K.sum(K.logical_and(y_true, y_pred))
den = K.sum(y_true)
return K.divide(num, den)
def f_measure(y_true, y_pred):
p = precision(y_true, y_pred)