wehub-resource-sync
14 lines
246 B
Python
Executable File
14 lines
246 B
Python
Executable File
import unittest
|
|
import numpy as np
|
|
from prml import nn
|
|
|
|
|
|
class TestSoftmax(unittest.TestCase):
|
|
|
|
def test_softmax(self):
|
|
self.assertTrue(np.allclose(nn.softmax(np.ones(4)).value, 0.25))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|