python 多分类 recall_多类CNN的宏度量(recall/F1…)

该博客讨论了在多类别不平衡数据集上使用CNN进行图像分类时,如何计算宏平均的F1分数、精度和召回率。作者遇到的问题是不同代码实现的宏调用结果不一致,并探讨了使用自定义召回函数和加权分类交叉熵的效果。
摘要由CSDN通过智能技术生成

我使用CNN对不平衡数据集进行图像分类。我对tensorflow后端是全新的。这是一个多类问题(不是多标签问题),我有16个类。类是一个热编码的。在

我想计算每个历元的宏指标:F1,精度和召回率。在class Metrics(Callback):

def on_train_begin(self, logs={}):

self.val_f1s = []

self.val_recalls = []

self.val_precisions = []

def on_epoch_end(self, epoch, logs={}):

val_predict = (np.asarray(self.model.predict(self.validation_data[0]))).round()

val_targ = self.validation_data[1]

_val_f1 = f1_score(val_targ, val_predict,average='macro')

_val_recall = recall_score(val_targ, val_predict,average='macro')

_val_precision = precision_score(val_targ, val_predict,average='macro')

self.val_f1s.append(_val_f1)

self.val_recalls.append(_val_recall)

self.val_precisions.append(_val_precision)

print (" — val_f1: %f — val_precision: %f — v

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值