这个屌丝很懒，什么也没留下！

Keras自定义损失函数和度量显示inf解决方法_keras loss inf

作者：凡人多烦事01 | 2024-04-01 20:40:56

keras loss inf

去掉自定义函数的axis=-1


def rmse(y_true, y_pred):
    return backend.sqrt(backend.mean(backend.square(y_pred - y_true)))
model_lstm.compile(optimizer="adam", loss=rmse, metrics=[rmse])

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/凡人多烦事01/article/detail/350045