Como escrever um loop de treinamento
No scikit-learn
, o loop de treinamento está incluído no método .fit()
, enquanto no PyTorch ele é configurado manualmente. Embora isso aumente a flexibilidade, requer uma implementação personalizada.
Neste exercício, você criará um loop para treinar um modelo de previsão de salário.
A função show_results()
é fornecida para ajudar você a visualizar algumas previsões de amostra.
As importações de pacotes fornecidas são: pandas como pd
, torch
, torch.nn
como nn
, torch.optim
como optim
, além de DataLoader
e TensorDataset
de torch.utils.data
.
Foram criadas as seguintes variáveis: num_epochs
, contendo o número de épocas (definido como 5); dataloader
, contendo o carregador de dados; model
, contendo a rede neural; criterion
, contendo a função de perda, nn.MSELoss()
; optimizer
, contendo o otimizador SGD.
Este exercício faz parte do curso
Introdução ao Aprendizado Profundo com o PyTorch
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Loop over the number of epochs and then the dataloader
for i in ____:
for data in ____:
# Set the gradients to zero
____