Sobes.tech
Middle

Kokią nuostolių funkciją taikyti modelio mokymo metu?

sobes.tech AI

Atsakymas iš AI

Mašininio mokymosi užduoties ir modelio tipo priklauso nuo prarandų funkcijos (loss funkcijos) pasirinkimas:

  • Regresijos užduotims dažnai naudojama Mean Squared Error (MSE) arba Mean Absolute Error (MAE).
  • Dviem klasėms skirtoms klasifikacijos užduotims dažnai taikoma Binary Cross-Entropy (logistinė regresija).
  • Daugklasės klasifikacijai — Categorical Cross-Entropy.
  • Užduotims su nesubalansuotomis klasėmis gali būti naudojamos svorines prarandų funkcijų versijos arba Focal Loss.

Pavyzdžiui, mokant neuroninį tinklą 10 klasių vaizdų klasifikacijai, dažnai naudojama Categorical Cross-Entropy:

loss = tf.keras.losses.CategoricalCrossentropy()

Svarbu pasirinkti prarandų funkciją, kuri atitinka kokybės matą ir modelio išėjimo duomenų tipą.