tnmf.tests.test_minibatch

Test the decomposition on multiple image patches in minibatch mode and verify that all backends yield the same factorization.

Module Contents

Functions

fit_nmf(backend, algorithm)
fixture_expected_factorization(algorithm)
test_expected_energy(backend: str, algorithm: int, expected_factorization: Tuple[numpy.ndarray, numpy.ndarray])

Attributes

expected_energies
backends
img
shape
image_shape
patch_shape
byte_size
image_strides
V
V

tnmf.tests.test_minibatch.expected_energies

tnmf.tests.test_minibatch.backends = ['numpy', 'numpy_fft', 'numpy_caching_fft', 'pytorch', 'pytorch_fft']

tnmf.tests.test_minibatch.img

tnmf.tests.test_minibatch.shape

tnmf.tests.test_minibatch.image_shape

tnmf.tests.test_minibatch.patch_shape

tnmf.tests.test_minibatch.byte_size

tnmf.tests.test_minibatch.image_strides

tnmf.tests.test_minibatch.V

tnmf.tests.test_minibatch.V

tnmf.tests.test_minibatch.fit_nmf(backend, algorithm)

tnmf.tests.test_minibatch.fixture_expected_factorization(algorithm)

tnmf.tests.test_minibatch.test_expected_energy(backend: str, algorithm: int, expected_factorization: Tuple[numpy.ndarray, numpy.ndarray])