Esempi in Python per SomContainer.get_bmu

Esempio n. 1

File: test_som_container.py Progetto: kamilk92/torchsom

 def test_provide_bmu_with_custim_dist(self):
     dist = MagicMock()
     bmu = SomContainer.get_bmu(torch.tensor(np.zeros((5, 5))), dist)
     self.assertIsInstance(bmu, Bmu)
     self.assertEqual(bmu._distance, dist)

Esempio n. 2

File: test_som_container.py Progetto: kamilk92/torchsom

 def test_provide_bmu_with_default_dist(self):
     bmu = SomContainer.get_bmu(torch.tensor(np.zeros((5, 5))))
     self.assertIsInstance(bmu, Bmu)