replaced np with K
This commit is contained in:
parent
92eb1dce75
commit
70fe296360
@ -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)
|
||||
|
Reference in New Issue
Block a user