keras:f1_score作为metric,存储模型, 加载带有自定义函数的模型(1/2)

本文主要关于如何在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,存储模型, 加载带有自定义函数的模型(1/2))