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

@ -48,21 +48,21 @@ def FCN():
## Define the model structure
model = Model(inputs=inputs, outputs=classif)
# Optimizer recommended Adadelta values (lr=0.01)
model.compile(optimizer=Adam(), loss='binary_crossentropy', metrics=['accuracy'])
model.compile(optimizer=Adam(), loss='binary_crossentropy', metrics=['accuracy'])
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)