在Keras中定义特定的自定义指标

时间:2020-07-10 09:15:15

标签: python tensorflow keras neural-network custom-object

我正在尝试在Python中定义一个我想用作度量标准的方法,尤其是对于EarlyStopping(restore_best_weights)。 问题是我正在尝试使用此方法(使用当前参数)进行预测,但似乎不起作用。 (我需要对特定的递归问题进行预测)

请参见以下简化示例:

import tensorflow as tf
from tensorflow.keras.layers import *
from tensorflow.keras.activations import *
from tensorflow.keras.models import *
from tensorflow.keras.optimizers import *
from tensorflow.keras.initializers import *
from tensorflow.keras.callbacks import *
import numpy as np

x_train = np.zeros((100, 7))
y_train = np.zeros(100)

model = Sequential()
model.add(Dense(units=7, input_shape=(x_train.shape[1], )))
model.add(Dense(units=1))
model.add(Activation('sigmoid'))

input1 = np.zeros((5, 7), dtype=np.float32)
y_hat = model.predict(input1)
print(y_hat)

def testMetric(y_true, y_pred):
    #input1 = np.zeros((5, 7), dtype=np.float32)
    #y_hat = model.predict(input1)
    return 5

model.compile(
loss="binary_crossentropy",
optimizer=Adam(0.05),
metrics=['binary_accuracy', testMetric]
)

reduce_lr = ReduceLROnPlateau(monitor='testMetric', min_delta=0, factor=0.7, patience=1, verbose=1, mode='max')
early = EarlyStopping(monitor='testMetric', min_delta=0, patience=7, verbose=1, mode='max', baseline=None, restore_best_weights=True)
model.fit(
    x=x_train,
    y=y_train,
    epochs=3,
    callbacks=[early, reduce_lr]
    )

当我在方法“ testMetric”中不使用预测时,一切都很好。但是当我使用谓词(取消注释)时,我收到一条错误消息。

RuntimeError: Method requires being in cross-replica context, use get_replica_context().merge_call()

是否可以在我的方法中使用预测?

我正在使用Tensorflow 2.2.0

那对我有帮助:)

0 个答案:

没有答案