[英]How to convert this loss function written in Lasagne to Keras?
我一直致力于将用Lasagne编写的CNN转换为Keras(链接到Lasagne版本: https : //github.com/MTG/DeepConvSep/blob/master/examples/dsd100/trainCNN.py )我相信我已经大部分都失败了,然而,损失函数是我努力重写的部分。
网络的输出具有形状(32,513,30,4),并且损失函数使用不同的层(最后的暗淡)。 我试图将其重写为自定义丢失函数,我可以将其插入到model.compile()这里是我编写的代码:
rand_num = np.random.uniform(size=(32,513,30,1))
epsilon=1e-8
alpha=0.001
beta=0.01
beta_voc=0.03
def loss_func(y_true, y_pred):
global alpha, beta, beta_voc, rand_num
voc = y_pred[:,:,:,0:1] + epsilon * rand_num
bass = y_pred[:,:,:,1:2] + epsilon * rand_num
dru = y_pred[:,:,:,2:3] + epsilon * rand_num
oth = y_pred[:,:,:,3:4] + epsilon * rand_num
mask_vox = voc/(voc+bass+dru+oth)
mask_bass = bass/(voc+bass+dru+oth)
mask_drums = dru/(voc+bass+dru+oth)
mask_oth = oth/(voc+bass+dru+oth)
vocals = mask_vox * inp
bass = mask_bass * inp
drums = mask_drums * inp
other = mask_oth * inp
train_loss_vocals = mean_squared_error(y_true=y_true[:,:,:,0:1],y_pred=vocals)
alpha_component = alpha*mean_squared_error(y_true=y_true[:,:,:,1:2],y_pred=vocals)
alpha_component += alpha*mean_squared_error(y_true=y_true[:,:,:,2:3],y_pred=vocals)
train_loss_recon_neg_voc = beta_voc*mean_squared_error(y_true=y_true[:,:,:,3:4],y_pred=vocals)
train_loss_bass = mean_squared_error(y_true=y_true[:,:,:,1:2],y_pred=bass)
alpha_component += alpha*mean_squared_error(y_true=y_true[:,:,:,0:1],y_pred=bass)
alpha_component += alpha*mean_squared_error(y_true=y_true[:,:,:,2:3],y_pred=bass)
train_loss_recon_neg = beta*mean_squared_error(y_true=y_true[:,:,:,3:4],y_pred=bass)
train_loss_drums = mean_squared_error(y_true=y_true[:,:,:,2:3],y_pred=drums)
alpha_component += alpha*mean_squared_error(y_true=y_true[:,:,:,0:1],y_pred=drums)
alpha_component += alpha*mean_squared_error(y_true=y_true[:,:,:,1:2],y_pred=drums)
train_loss_recon_neg += beta*mean_squared_error(y_true=y_true[:,:,:,3:4],y_pred=drums)
vocals_error=train_loss_vocals.sum()
drums_error=train_loss_drums.sum()
bass_error=train_loss_bass.sum()
negative_error=train_loss_recon_neg.sum()
negative_error_voc=train_loss_recon_neg_voc.sum()
alpha_component=alpha_component.sum()
loss=abs(vocals_error+drums_error+bass_error-negative_error-alpha_component-negative_error_voc)
return loss
我得到的第一个错误是:
AttributeError:'Tensor'对象没有属性'sum'
但是,我不确定其他一些操作,即使是不正确的操作,也会引发错误。
我真的很感激一些帮助。 谢谢。
例如,请参阅此示例,了解如何在Keras中定义自定义丢失。
y_pred
是y_pred
模型的输出是张量。 你必须调整你的操作以使用张量,所以你不得不使用K.sum
而不是np.sum
之类的东西,而K是你的keras后端。
有些运算符过载,因此您可以使用+
添加两个张量。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.