[心得] torchnmf -- 用PyTorch寫非負矩陣分解
[關鍵字]: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
推
11/10 20:48,
6年前
, 1F
11/10 20:48, 1F
推
11/11 00:45,
6年前
, 2F
11/11 00:45, 2F
穴穴學長~
※ 編輯: yoyololicon (1.163.58.153), 11/11/2018 09:33:36
推
01/19 21:29,
6年前
, 3F
01/19 21:29, 3F
DataScience 近期熱門文章
PTT數位生活區 即時熱門文章