keras自定义函数

metrics

从网上瞎找了个代码改过来的，然后看了看from sklearn.metrics import r2_score的代码，和我的一样，放心了。

import keras.backend as K
def keras_r2(y_true,y_pred):
    y_mean=K.mean(y_true)
    # ssreg=K.sum((y_pred-y_mean)**2)
    sstotal=K.sum((y_true-y_mean)**2)
    ssres=K.sum((y_true-y_pred)**2)
    score = 1-(ssres/sstotal)
    return score