def dice_core(y, y_pred):
y_core = np.zeros(y.shape)
y_core[np.where( (y==1) | (y==3) | (y==4) )[0]] = 1
y_pred_core = np.zeros(y_pred.shape)
y_pred_core[np.where( (y_pred==1) | (y_pred==3) | (y_pred==4) )[0] ] = 1
return (2. * np.sum(y_pred_core*y_core) + np.finfo(float).eps) / (np.sum(y_core) + np.sum(y_pred_core)+ np.finfo(float).eps)
评论列表
文章目录