ComeçarComece de graça

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

Ver curso

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
    ____
Editar e executar o código