def testCheckRegularizers(self):
regularizers = {
"key_a": tf.contrib.layers.l1_regularizer(scale=0.5),
"key_c": tf.contrib.layers.l2_regularizer(scale=0.5)
}
keys = ["key_a", "key_b"]
self.assertRaisesRegexp(KeyError,
"Invalid regularizer keys.*",
snt.check_regularizers,
regularizers=regularizers,
keys=keys)
del regularizers["key_c"]
regularizers["key_b"] = "not a function"
self.assertRaisesRegexp(TypeError,
"Regularizer for.*",
snt.check_regularizers,
regularizers=regularizers,
keys=keys)
regularizers["key_b"] = {"key_c": "not a function"}
self.assertRaisesRegexp(TypeError,
"Regularizer for.*",
snt.check_regularizers,
regularizers=regularizers,
keys=keys)
regularizers["key_b"] = {
"key_c": tf.contrib.layers.l1_regularizer(scale=0.5),
"key_d": tf.contrib.layers.l2_regularizer(scale=0.5)
}
snt.check_regularizers(regularizers=regularizers, keys=keys)
def testCheckRegularizers(self):
regularizers = {
"key_a": tf.contrib.layers.l1_regularizer(scale=0.5),
"key_c": tf.contrib.layers.l2_regularizer(scale=0.5),
}
keys = ["key_a", "key_b"]
self.assertRaisesRegexp(KeyError,
"Invalid regularizer keys.*",
snt.check_regularizers,
regularizers=regularizers,
keys=keys)
del regularizers["key_c"]
regularizers["key_b"] = "not a function"
self.assertRaisesRegexp(TypeError,
"Regularizer for.*",
snt.check_regularizers,
regularizers=regularizers,
keys=keys)
regularizers["key_b"] = {"key_c": "not a function"}
self.assertRaisesRegexp(TypeError,
"Regularizer for.*",
snt.check_regularizers,
regularizers=regularizers,
keys=keys)
regularizers["key_b"] = {
"key_c": tf.contrib.layers.l1_regularizer(scale=0.5),
"key_d": tf.contrib.layers.l2_regularizer(scale=0.5),
}
snt.check_regularizers(regularizers=regularizers, keys=keys)
def __init__(self,
output_size,
use_bias=True,
initializers=None,
partitioners=None,
regularizers=None,
custom_getter=None,
name="embed_lin"):
super(AbstractGraphLayer, self).__init__(custom_getter=None, name=name)
self._output_size = output_size
self._use_bias = use_bias
self._input_shape = None
self.possible_keys = self.get_possible_initializer_keys(
use_bias=use_bias)
self._initializers = snt.check_initializers(initializers,
self.possible_keys)
self._partitioners = snt.check_partitioners(partitioners,
self.possible_keys)
self._regularizers = snt.check_regularizers(regularizers,
self.possible_keys)
