Začněte nyníZačněte zdarma

Psaní trénovací smyčky

V scikit-learn je trénovací smyčka zabalena v metodě .fit(), zatímco v PyTorchi se nastavuje ručně. To sice přináší větší flexibilitu, ale zároveň vyžaduje vlastní implementaci.

V tomto cvičení vytvoříš smyčku pro trénování modelu předpovídajícího platy.

Funkce show_results() je k dispozici, aby ti pomohla vizualizovat ukázkové předpovědi.

Dostupné importy jsou: pandas jako pd, torch, torch.nn jako nn, torch.optim jako optim a také DataLoader a TensorDataset z torch.utils.data.

Byly vytvořeny následující proměnné: num_epochs s počtem epoch (nastaveným na 5); dataloader s dataloaderem; model s neuronovou sítí; criterion s účelovou funkcí nn.MSELoss(); optimizer s SGD optimizerem.

Toto cvičení je součástí kurzu

Introduction to Deep Learning with PyTorch

Zobrazit kurz

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

# Loop over the number of epochs and then the dataloader
for i in ____:
  for data in ____:
    # Set the gradients to zero
    ____
Upravit a spustit kód