def test_evaluate_matrix(self):
mat_in = np.array([
[0, 0, 1, 0, 0, 1],
[1, 0, 0, 1, 0, 0],
[0, 0, 0, 1, 1, 0],
])
mat_out = np.array([
[1, 0, 1, 0],
[1, 0, 0, 1],
[0, 1, 0, 1],
])
binam = pynam.binam.BiNAM(6, 4)
binam.train_matrix(mat_in, mat_out)
mat_out_recall = binam.evaluate_matrix(mat_in)
np.testing.assert_equal(mat_out_recall, mat_out)
def test_train_matrix(self):
mat_in = np.array([
[0, 0, 1, 0, 0, 1],
[1, 0, 0, 1, 0, 0],
[0, 0, 0, 1, 1, 0],
])
mat_out = np.array([
[1, 0, 1, 0],
[1, 0, 0, 1],
[0, 1, 0, 1],
])
binam = pynam.binam.BiNAM(6, 4)
binam.train_matrix(mat_in, mat_out)
np.testing.assert_equal([
[1, 0, 0, 1],
[0, 0, 0, 0],
[1, 0, 1, 0],
[1, 1, 0, 1],
[0, 1, 0, 1],
[1, 0, 1, 0]
], binam.get())
