wehub-resource-sync
28 lines
624 B
Python
Executable File
28 lines
624 B
Python
Executable File
import unittest
|
|
import numpy as np
|
|
from prml import nn
|
|
|
|
|
|
class TestDeterminant(unittest.TestCase):
|
|
|
|
def test_determinant(self):
|
|
A = np.array([
|
|
[2., 1.],
|
|
[1., 3.]
|
|
])
|
|
detA = np.linalg.det(A)
|
|
self.assertTrue((detA == nn.linalg.det(A).value).all())
|
|
|
|
A = nn.Parameter(A)
|
|
for _ in range(100):
|
|
A.cleargrad()
|
|
detA = nn.linalg.det(A)
|
|
loss = nn.square(detA - 1)
|
|
loss.backward()
|
|
A.value -= 0.1 * A.grad
|
|
self.assertAlmostEqual(detA.value, 1.)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|