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
____