tf.keras.losses.get

Возвращает функцию потерь Keras в виде экземпляра класса function/Loss.

tf.keras.losses.get(
    identifier
)

Идентификатор может быть именем функции потерь в виде строки или экземпляром класса Loss.

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'>

Вы также можете указать параметры функции потерь, передав словарь, содержащий class_name и config в качестве идентификатора. Также обратите внимание, что class_name должен соответствовать классу Loss.

identifier = {"class_name": "CategoricalCrossentropy",
              "config": {"from_logits": True} }
loss = tf.keras.losses.get(identifier)
type(loss)
<class '...tensorflow.python.keras.losses.CategoricalCrossentropy'>