50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
|
from keras_core import testing
|
||
|
|
from keras_core import initializers
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
|
||
|
|
class InitializersTest(testing.TestCase):
|
||
|
|
def test_random_normal(self):
|
||
|
|
shape = (5, 5)
|
||
|
|
mean = 0.0
|
||
|
|
stddev = 1.0
|
||
|
|
seed = 1234
|
||
|
|
external_config = {"mean": 1.0, "stddev": 0.5, "seed": 42}
|
||
|
|
initializer = initializers.RandomNormal(
|
||
|
|
mean=mean,
|
||
|
|
stddev=stddev,
|
||
|
|
seed=seed
|
||
|
|
)
|
||
|
|
values = initializer(shape=shape)
|
||
|
|
self.assertEqual(initializer.mean, mean)
|
||
|
|
self.assertEqual(initializer.stddev, stddev)
|
||
|
|
self.assertEqual(initializer.seed, seed)
|
||
|
|
self.assertEqual(values.shape, shape)
|
||
|
|
self.test_load_external_config(initializer, external_config)
|
||
|
|
|
||
|
|
def test_random_uniform(self):
|
||
|
|
shape = (5, 5)
|
||
|
|
minval = -1.0
|
||
|
|
maxval = 1.0
|
||
|
|
seed = 1234
|
||
|
|
external_config = {"minval": 0.0, "maxval": 1.0, "seed": 42}
|
||
|
|
initializer = initializers.RandomUniform(
|
||
|
|
minval=minval,
|
||
|
|
maxval=maxval,
|
||
|
|
seed=seed
|
||
|
|
)
|
||
|
|
values = initializer(shape=shape)
|
||
|
|
self.assertEqual(initializer.minval, minval)
|
||
|
|
self.assertEqual(initializer.maxval, maxval)
|
||
|
|
self.assertEqual(initializer.seed, seed)
|
||
|
|
self.assertEqual(values.shape, shape)
|
||
|
|
self.test_load_external_config(initializer, external_config)
|
||
|
|
values = values.numpy()
|
||
|
|
self.assertGreaterEqual(np.min(values), minval)
|
||
|
|
self.assertLess(np.max(values), maxval)
|
||
|
|
|
||
|
|
def test_load_external_config(self, initializer, config):
|
||
|
|
initializer = initializer.from_config(config)
|
||
|
|
self.assertEqual(initializer.get_config(), config)
|
||
|
|
|