美文网首页
keras:f1_score作为metric,存储模型, 加载带

keras:f1_score作为metric,存储模型, 加载带

作者: 泡泡_e661 | 来源:发表于2019-11-05 04:58 被阅读0次

本文主要关于如何在keras模型编译中,使用自定义函数作为metric对模型进行评价。

前提:不平衡数据,想使用f1_score作为模型的metric评价模型的好坏。使用Keras框架构建模型。

本文适合所以自定义的metric情况,在这里自定义了f1_score函数,函数来源于https://datascience.stackexchange.com/questions/45165/how-to-get-accuracy-f1-precision-and-recall-for-a-keras-model

# compile the model

model.compile(optimizer='adam', loss='binary_crossentropy', metrics=[f1_m])

相关文章

网友评论

      本文标题:keras:f1_score作为metric,存储模型, 加载带

      本文链接:https://www.haomeiwen.com/subject/zfnjbctx.html