PyTorch NMF Documentation

PyTorch NMF is a extension library for PyTorch.

It consists of basic NMF algorithm and its convolutional variants, which are hardly found in other NMF packages. In addition, by using the PyTorch automatic differentiation feature, it is able to adopt the classic multiplicative update rules into more complex NMF structures, and make it possible to train these complex models in a simple end-to-end fashion.

Notes

• Installation
• Introduction by Example

Package Reference

• torchnmf.nmf
• torchnmf.plca
• torchnmf.metrics
• torchnmf.trainer

Indices and tables

• Index

• Module Index