这是正确的相关系数keras度量吗?

时间:2019-07-08 15:32:46

标签: deep-learning

我想问一下以下代码是否可以正确计算相关系数,因为我想将其用作 keras指标?

def correlation_coefficient(y_true, y_pred):
    pearson_r, update_op = tf.contrib.metrics.streaming_pearson_correlation(y_pred, y_true, name='pearson_r')
    metric_vars = [i for i in tf.local_variables() if 'pearson_r'  in i.name.split('/')]

    for v in metric_vars:
        tf.add_to_collection(tf.GraphKeys.GLOBAL_VARIABLES, v)

    with tf.control_dependencies([update_op]):
        pearson_r = tf.identity(pearson_r)
        return 1-pearson_r**2

0 个答案:

没有答案