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

cvičení

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.

Pokyny 1/3

undefined XP
    1
    2
    3
  • Napiš for smyčku, která iteruje přes dataloader; ta by měla být vnořena do for smyčky iterující přes rozsah odpovídající počtu epoch.
  • Nastav gradienty optimizeru na nulu.