1. Nauka
    2. /
  2. Kursy
    3. /
  3. Wprowadzenie do uczenia głębokiego z PyTorch
Connected

ćwiczenie

Eksperymenty z dropout

Dropout pomaga zapobiegać przeuczeniu, losowo zerując część wartości wyjściowych podczas treningu. W tym ćwiczeniu zbudujesz prostą sieć neuronową z dropoutem i sprawdzisz, jak zachowuje się w trybie treningu i ewaluacji.

Pakiet torch.nn jest wstępnie załadowany jako nn, a zmienna features jest już dla ciebie zdefiniowana.

Instrukcje 1/2

undefined XP
    1
    2
  • Dodaj do modelu warstwę dropout z 50% prawdopodobieństwem zerowania elementów.
  • Ustaw model w trybie treningu i wygeneruj predykcje na podstawie features, zapisując wynik jako output_train.