replaced np with K
This commit is contained in:
parent
92eb1dce75
commit
70fe296360
@ -53,16 +53,16 @@ def FCN():
|
|||||||
return model
|
return model
|
||||||
|
|
||||||
def precision(y_true, y_pred):
|
def precision(y_true, y_pred):
|
||||||
y_pred = np.round(y_pred)
|
y_pred = K.round(y_pred)
|
||||||
num = np.sum(np.logical_and(y_true, y_pred))
|
num = K.sum(K.logical_and(y_true, y_pred))
|
||||||
den = np.sum(y_pred)
|
den = K.sum(y_pred)
|
||||||
return np.divide(num, den)
|
return K.divide(num, den)
|
||||||
|
|
||||||
def recall(y_true, y_pred):
|
def recall(y_true, y_pred):
|
||||||
y_pred = np.round(y_pred)
|
y_pred = K.round(y_pred)
|
||||||
num = np.sum(np.logical_and(y_true, y_pred))
|
num = K.sum(K.logical_and(y_true, y_pred))
|
||||||
den = np.sum(y_true)
|
den = K.sum(y_true)
|
||||||
return np.divide(num, den)
|
return K.divide(num, den)
|
||||||
|
|
||||||
def f_measure(y_true, y_pred):
|
def f_measure(y_true, y_pred):
|
||||||
p = precision(y_true, y_pred)
|
p = precision(y_true, y_pred)
|
||||||
|
Reference in New Issue
Block a user