Jakou ztrátovou funkci použít při tréninku modelu?

Výběr ztrátové funkce (loss funkce) závisí na úkolu strojového učení a typu modelu:

• Pro regresní úkoly se obvykle používá Mean Squared Error (MSE) nebo Mean Absolute Error (MAE).
• Pro binární klasifikační úkoly se často používá Binary Cross-Entropy (logistická regrese).
• Pro vícekategoriální klasifikaci — Categorical Cross-Entropy.
• Pro úkoly s nevyváženými třídami lze použít vážené verze ztrátových funkcí nebo Focal Loss.

Například při tréninku neuronové sítě pro klasifikaci obrázků s 10 třídami se obvykle používá Categorical Cross-Entropy:

loss = tf.keras.losses.CategoricalCrossentropy()

Je důležité zvolit ztrátovou funkci, která odpovídá metrice kvality a typu výstupních dat modelu.