栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Keras加入F1,recall,precision评价指标

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Keras加入F1,recall,precision评价指标

按照Keras给出的一些代码,评价指标通常只有ACC跟loss,如果要看F1等其他参数就看不了,有跟多博客给出了添加的方式,大多是自己手写了公式,然后添加到里面去,个人感觉这种方式并不稳妥,给出我用的方法,直接调用官方给出的库函数即可。

import keras_metrics as km

然后在
model.compile中添加即可。给出一个LSTM的例子:

def BILSTM_model(self,word_index,embeddings_index):
        print('开始构建BILSTM模型')
        embedding_layer = self.emb_model_layer(word_index,embeddings_index)
        sequence_input = Input(shape=(MAX_SEQUENCE_LENGTH,), dtype='int32')
        embedded_sequences = embedding_layer(sequence_input)
        l_1 = Bidirectional(LSTM(100, return_sequences=False))(embedded_sequences)
        dense_1 = Dense(100,activation='tanh')(l_1)
        dense_2 = Dense(2, activation='softmax')(dense_1)

        model = Model(sequence_input, dense_2)

        model.compile(loss='categorical_crossentropy',
                  optimizer='rmsprop',
                  metrics=['acc',km.f1_score(), km.binary_precision(), km.binary_recall()])//在这个地方添加需要的参数

        model.summary()
        return model
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/754052.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号