pytorch-nmf实现

2022-12-02 21:20 作者:021usc 0人读过 | 我要投稿

参考

https://yoyololicon.github.io/posts/2021/02/torchnmf-algorithm/

https://github.com/yoyololicon/pytorch-NMF

https://gitee.com/youryouth/mc/tree/master/nmf

这里是通过下载别人写好的库实现的。首先安装对应的库

然后通过下述代码，实现单层的矩阵分解

上图左为原始图片，上图右为矩阵分解后补全的图片，可以看出分解后的图片在某些细节处理上并不是那么好。

将矩阵 $X$ 进行深度分解，即 $X%20%3D%20W_1W_2H_2$ 。

标签：