wehub-resource-sync
22 lines
547 B
Python
Executable File
22 lines
547 B
Python
Executable File
import unittest
|
|
import numpy as np
|
|
from prml import nn
|
|
|
|
|
|
class TestExponential(unittest.TestCase):
|
|
|
|
def test_exponential(self):
|
|
np.random.seed(1234)
|
|
obs = np.random.gamma(1, 1 / 0.5, size=1000)
|
|
a = nn.Parameter(0)
|
|
for _ in range(100):
|
|
a.cleargrad()
|
|
x = nn.random.Exponential(nn.softplus(a), data=obs)
|
|
x.log_pdf().sum().backward()
|
|
a.value += a.grad * 0.001
|
|
self.assertAlmostEqual(x.rate.value, 0.475135117)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|