def __call__(self, x, pcaed_x=None, test=False):
h = self.enc_l(x)
rec = self.dec_l(h)
if test:
return rec, h
ae_loss = F.mean_squared_error(x, rec)
pca_loss = F.mean_squared_error(pcaed_x, h) * self.c
# self.loss = ae_loss + pca_loss
self.loss = pca_loss
chainer.reporter.report({'loss': self.loss,
'ae_loss': ae_loss,
'pca_loss': pca_loss,
}, self)
return self.loss
my_pca.py 文件源码
python
阅读 21
收藏 0
点赞 0
评论 0
评论列表
文章目录