Keras 利用sklearn的ROC-AUC建立评价函数

柔光的暖阳◎ 2021-06-24 15:58 411阅读 0赞

# 利用sklearn自建评价函数
    from sklearn.model_selection import train_test_split
    from sklearn.metrics import roc_auc_score
    from keras.callbacks import Callback
    
    class RocAucEvaluation(Callback):
        def __init__(self, validation_data=(), interval=1):
            super(Callback, self).__init__()
            self.interval = interval
            self.x_val,self.y_val = validation_data
        def on_epoch_end(self, epoch, log={}):
            if epoch % self.interval == 0:
                y_pred = self.model.predict(self.x_val, verbose=0)
                score = roc_auc_score(self.y_val, y_pred)
                print('\n ROC_AUC - epoch:%d - score:%.6f \n' % (epoch+1, score))
    
    x_train,y_train,x_label,y_label = train_test_split(train_feature, train_label, train_size=0.95, random_state=233)
    RocAuc = RocAucEvaluation(validation_data=(y_train,y_label), interval=1)
    
    
    hist = model.fit(x_train, x_label, batch_size=batch_size, epochs=epochs, validation_data=(y_train, y_label), callbacks=[RocAuc], verbose=2)

发表评论取消回复

表情：

评论列表（有 0 条评论，411人围观）

还没有评论，来说两句吧...

相关阅读

相关利用sklearn对多分类的每个类别进行指标评价

今天晚上，笔者接到客户的一个需要，那就是：对多分类结果的每个类别进行指标评价，也就是需要输出每个类型的精确率（precision），召回率（recall）以及F1值（F1-

缺乏、安全感/ 2023年06月13日 06:29/ 0 赞/ 57 阅读

相关使用sklearn对多分类的每个类别进行指标评价操作

使用sklearn对多分类的每个类别进行指标评价操作更多python视频教程请到菜鸟教程https://www.piaodoo.com/ 今天晚上，笔者接到客户的一个需要

灰太狼/ 2023年02月19日 06:25/ 0 赞/ 21 阅读

相关评价函数_案例 | Excel九宫格矩阵评价统计，离不开这些函数

营长说在绩效管理方案中，经常会遇到矩阵评价，常见的有2\2和3\3矩阵。总有学员问与矩阵查询统计的问题，有一定的代表性。今天营长就以经典的3\3九宫格矩阵为例，向你介绍涉及

川长思鸟来/ 2023年01月06日 15:53/ 0 赞/ 121 阅读

相关 sklearn的评价函数

参考资料：[https://www.cnblogs.com/wj-1314/p/9400375.html][https_www.cnblogs.com_wj-1314_p_94

客官°小女子只卖身不卖艺/ 2022年12月21日 13:59/ 0 赞/ 162 阅读

相关基于sklearn和keras的数据切分与交叉验证

目录 [一．自动切分][Link 1] [二．手动切分][Link 2] [三．K折交叉验证（k-fold cross validation）][K

清疚/ 2022年10月29日 03:22/ 0 赞/ 173 阅读

相关 LSTM函数详解（keras）

原文链接：[https://www.zhihu.com/question/64470274][https_www.zhihu.com_question_64470274] [

约定不等于承诺〃/ 2022年09月12日 15:53/ 0 赞/ 196 阅读

相关 Keras loss函数

''' Created on 2018-4-16 ''' def compile( self, optimizer, 优化器

╰半夏微凉°/ 2022年05月27日 09:13/ 0 赞/ 231 阅读

相关 Keras：评价指标 categorical_accuracy和 sparse_categorical_accuracy

在Keras中，官方内置了几种评价函数。对于二分类问题，评价指标可以用 binary\_accuracy，就是最直观上讲的准确率。当面对多分类或者多标签的任务

谁借莪１个温暖的怀抱￠/ 2022年04月24日 01:48/ 0 赞/ 285 阅读

相关 sklearn函数记录

记录用到的一些sklearn的函数~ 文章目录 1.关于CountVectorizer 2.关于TfidfVect

Myth丶恋晨/ 2022年01月27日 05:41/ 0 赞/ 259 阅读

相关 Keras 利用sklearn的ROC-AUC建立评价函数

利用sklearn自建评价函数 from sklearn.model_selection import train_test_split from s

柔光的暖阳◎/ 2021年06月24日 15:58/ 0 赞/ 412 阅读