def multilabel_loss(preds, labels):
eps = 1e-4
preds = T.clip(preds, eps, 1-eps)
return -(labels * T.log(preds) + (1 - labels) * T.log(1 - preds)).mean(axis=1).mean(axis=0)
微信
公众号
扫码关注公众号