python - 具有 tensorflow 功能的自定义 keras 指标。未初始化的变量

Question

我在 tensorflow 1.14 中运行具有自定义指标的模型，但我遇到了未初始化变量的问题。

def metric(y_true, y_pred):
    # some math operations all in tensorflow
    # ... 
    return tf.metrics.mean(tf.reduce_sum(my_mat, reduction_indices=[1]) / num_rel)

pred = tf.placeholder(dtype=tf.int32, shape=[None, 1]) 
model.compile(optimizer='adam', loss='mean_squared_error', metrics=[metric], target_tensors=pred)
met.test_on_batch(x, y)

这些错误是在指标中提出的，我希望我只是遗漏了一些简单的东西。任何帮助，将不胜感激。

标签： pythontensorflowkeraskeras-metrics