首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >如何在keras中计算接收工作特性(ROC)和AUC?

如何在keras中计算接收工作特性(ROC)和AUC?
EN

Stack Overflow用户
提问于 2016-12-08 13:44:46
回答 6查看 74.1K关注 0票数 63

我有一个多输出(200)二进制分类模型,它是我用keras编写的。

在这个模型中,我想添加额外的指标,比如ROC和AUC,但在我的知识内核中没有内置的ROC和AUC指标函数。

我试着从scikit-learn导入ROC,AUC函数

代码语言:javascript
复制
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'])

但是它给出了这个错误:

代码语言:javascript
复制
Exception: Invalid metric: roc_curve

如何将ROC,AUC添加到keras中?

EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/41032551

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档