本文主要关于如何在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])
网友评论