Retrieves a Keras loss as a function/Loss class instance.
tf.keras.losses.get(
identifier
)
The identifier may be the string name of a loss function or Loss class.
loss = tf.keras.losses.get("categorical_crossentropy")
type(loss)
<class 'function'>
loss = tf.keras.losses.get("CategoricalCrossentropy")
type(loss)
<class '...tensorflow.python.keras.losses.CategoricalCrossentropy'>
You can also specify config of the loss to this function by passing dict
containing class_name and config as an identifier. Also note that the
class_name must map to a Loss class
identifier = {"class_name": "CategoricalCrossentropy",
"config": {"from_logits": True} }
loss = tf.keras.losses.get(identifier)
type(loss)
<class '...tensorflow.python.keras.losses.CategoricalCrossentropy'>
Arguments |
|---|
identifier
|
A loss identifier. One of None or string name of a loss
function/class or loss configuration dictionary or a loss function or a
loss class instance
|
Returns |
|---|
A Keras loss as a function/ Loss class instance.
|
Raises |
|---|
ValueError
|
If identifier cannot be interpreted.
|
TensorFlow 1 version
View source on GitHub