How to compute Receiving Operating Characteristic (ROC) and AUC in keras?

前端 未结 8 1821
一生所求
一生所求 2020-11-28 02:07

I have a multi output(200) binary classification model which I wrote in keras.

In this model I want to add additional metrics such as ROC and AUC but to my knowledg

8条回答
  •  夕颜
    夕颜 (楼主)
    2020-11-28 02:52

    Adding to above answers, I got the error "ValueError: bad input shape ...", so I specify the vector of probabilities as follows:

    y_pred = model.predict_proba(x_test)[:,1]
    auc = roc_auc_score(y_test, y_pred)
    print(auc)
    

提交回复
热议问题