Home › Forums › Fórum Deep Learning de A à Z com PyTorch e Python › Atualização na classe da rede neural Projeto – 2: BreastCancer › Reply To: Atualização na classe da rede neural Projeto – 2: BreastCancer
13 de janeiro de 2022 at 15:03
#32905
Olá,
Esta atualização foi feita para que o modelo possa ser processado pela biblioteca Skorch, que não aceita que a saída da rede seja o resultado de uma ativação, mas sim exige que seja o resultado bruto. Neste caso, a função de custo fica responsável por fazer a ativação correspondente.
