ÎncepețiÎncepe gratuit

Ponderarea pierderii

Au fost antrenate trei versiuni ale modelului cu două ieșiri pentru predicția alfabetului și a caracterelor, pe care l-ai construit anterior: model_a, model_b și model_c. Pentru toate trei, pierderea a fost definită astfel:

loss_alpha = criterion(outputs_alpha, labels_alpha)
loss_char = criterion(outputs_char, labels_char)
loss = ((1 - char_weight) * loss_alpha) + (char_weight * loss_char)

Însă fiecare dintre cele trei modele a fost antrenat cu o valoare diferită pentru char_weight: 0.1, 0.5 sau 0.9.

Folosește funcția pe care ai definit-o în exercițiul anterior, evaluate_model(), pentru a verifica acuratețea fiecărui model. Care char_weight a fost folosit pentru antrenarea fiecărui model?

Acest exercițiu face parte din cursul

Deep Learning intermediar cu PyTorch

Vezi cursul

Exercițiu interactiv practic

Transformă teoria în practică cu unul dintre exercițiile noastre interactive

Începe exercițiul