Keras自定义RMSLE指标

时间:2017-11-30 21:34:42

标签: python deep-learning keras metrics

如何在Keras中实施此指标?我的代码下面给出了错误的结果! 请注意,我正在通过exp(x) - 1撤消先前的log(x + 1)变换,负面预测也会被剪切为0:

def rmsle_cust(y_true, y_pred):
    first_log = K.clip(K.exp(y_pred) - 1.0, 0, None)
    second_log = K.clip(K.exp(y_true) - 1.0, 0, None)
    return K.sqrt(K.mean(K.square(K.log(first_log + 1.) - K.log(second_log + 1.)), axis=-1)

为了比较,这是标准的numpy实现:

def rmsle_cust_py(y, y_pred, **kwargs):
    # undo 1 + log
    y = np.exp(y) - 1
    y_pred = np.exp(y_pred) - 1

    y_pred[y_pred < 0] = 0.0
    to_sum = [(math.log(y_pred[i] + 1) - math.log(y[i] + 1)) ** 2.0 for i,pred in enumerate(y_pred)]
    return (sum(to_sum) * (1.0/len(y))) ** 0.5

我做错了什么?谢谢!

编辑:设置axis=0似乎给出了一个非常接近正确值的值,但我不确定,因为我似乎使用了所有代码axis=-1

2 个答案:

答案 0 :(得分:7)

我遇到了同样的问题并搜索了它,这是我找到的

https://www.kaggle.com/jpopham91/rmlse-vectorized

经过修改后,这似乎对我有用,rmsle_K方法通过KerasTensorFlow实现。

import numpy as np
import math
from keras import backend as K
import tensorflow as tf

def rmsle(y, y0):
    assert len(y) == len(y0)
    return np.sqrt(np.mean(np.power(np.log1p(y)-np.log1p(y0), 2)))

def rmsle_loop(y, y0):
    assert len(y) == len(y0)
    terms_to_sum = [(math.log(y0[i] + 1) - math.log(y[i] + 1)) ** 2.0 for i,pred in enumerate(y0)]
    return (sum(terms_to_sum) * (1.0/len(y))) ** 0.5

def rmsle_K(y, y0):
    return K.sqrt(K.mean(K.square(tf.log1p(y) - tf.log1p(y0))))

r = rmsle(y=[5, 20, 12], y0=[8, 16, 12])
r1 = rmsle_loop(y=[5, 20, 12], y0=[8, 16, 12])
r2 = rmsle_K(y=[5., 20., 12.], y0=[8., 16., 12.])

print(r)

print(r1)

sess = tf.Session()

print(sess.run(r2))

<强>结果:

使用TensorFlow后端

0.263978210565

0.263978210565

0.263978

答案 1 :(得分:3)

通过在numpy实现中使用列表(to_sum),我怀疑你的numpy数组的形状为(length,)

在Keras上,由于axis=0axis=1得到了不同的结果,你可能会得到像(length,1)这样的形状。

此外,在创建to_sum列表时,您正在使用y[i]y_pred[i],这意味着您正在使用axis=0中的元素numpy implementation。

numpy实现还总结了计算sum(to_sum)中均值的所有内容。因此,您真的不需要使用axis中的任何K.mean

如果您确定模型的输出形状为(length,)(length,1),则可以仅使用K.mean(value)而不传递轴参数。