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