作者yoyololicon (萝莉大好)
看板DataScience
标题[心得] torchnmf -- 用PyTorch写非负矩阵分解
时间Sat Nov 10 17:13:17 2018
[关键字]:Non-negative Matrix Factorization
[重点摘要]:
安安大家毫
最近看了一些source seperation的paper,对於其中的NMF演算法觉得很有趣
就用pytorch写了一个算NMF的package,分享给大家~
可以用GPU加速,让原本就很快的NMF变得更快XD
https://i.imgur.com/hbSE6x0.png
另外也写了convolutional的NMF,好像没其他package有做convolutional的
(在pytorch里可以用conv1d简单的算出来
不过现在用NMF的人好像不多惹,都在搞DL QQ
https://github.com/yoyololicon/pytorch-NMFs
--
※ 发信站: 批踢踢实业坊(ptt.cc), 来自: 1.163.58.153
※ 文章网址: https://webptt.com/cn.aspx?n=bbs/DataScience/M.1541841202.A.408.html
1F:推 yougigun: 推! 11/10 20:48
2F:推 sma1033: 有贡献有推~ 11/11 00:45
穴穴学长~
※ 编辑: yoyololicon (1.163.58.153), 11/11/2018 09:33:36
3F:推 KuChanTung: 对降维相当有帮助 01/19 21:29