1. Učit se
  2. /
  3. Kurzy
  4. /
  5. Introduction to Deep Learning with PyTorch

Connected

cvičení

Výpočet cross-entropy loss

Cross-entropy loss je široce používaná metoda pro měření klasifikační ztráty. V tomto cvičení ji vypočítáš v PyTorchi pomocí:

  • y: skutečného popisku (ground truth label).
  • scores: vektoru predikcí před softmaxem.

Ztrátové funkce pomáhají neuronovým sítím učit se tím, že měří chyby predikcí. Vytvoř one-hot enkódovaný vektor pro y, definuj ztrátovou funkci cross-entropy a vypočítej ztrátu pomocí scores a enkódovaného popisku. Výsledkem bude jediné desetinné číslo reprezentující ztrátu pro daný vzorek.

torch, CrossEntropyLoss a torch.nn.functional jako F jsou již naimportované.

Pokyny 1/3

undefined XP
    1
    2
    3
  • Vytvoř one-hot enkódovaný vektor skutečného popisku y se 4 příznaky (jeden pro každou třídu) a přiřaď ho do proměnné one_hot_label.