作者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://www.ptt.cc/bbs/DataScience/M.1541841202.A.408.html
推 yougigun: 推! 11/10 20:48
推 sma1033: 有貢獻有推~ 11/11 00:45
穴穴學長~
※ 編輯: yoyololicon (1.163.58.153), 11/11/2018 09:33:36
推 KuChanTung: 對降維相當有幫助 01/19 21:29