我有一个多输出(200)二进制分类模型,它是我用keras编写的。
在这个模型中,我想添加额外的指标,比如ROC和AUC,但在我的知识内核中没有内置的ROC和AUC指标函数。
我试着从scikit-learn导入ROC,AUC函数
from sklearn.metrics import roc_curve, auc
from keras.models import Sequential
from keras.layers import Dense
.
.
.
model.add(Dense(200, activation='relu'))
model.add(Dense(300, activation='relu'))
model.add(Dense(400, activation='relu'))
model.add(Dense(300, activation='relu'))
model.add(Dense(200,init='normal', activation='softmax')) #outputlayer
model.compile(loss='categorical_crossentropy', optimizer='adam',metrics=['accuracy','roc_curve','auc'])
但是它给出了这个错误:
Exception: Invalid metric: roc_curve
如何将ROC,AUC添加到keras中?
https://stackoverflow.com/questions/41032551
复制相似问题