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

ćwiczenie

Funkcje sigmoid i softmax

Funkcje sigmoid i softmax to kluczowe funkcje aktywacji w uczeniu głębokim, często stosowane jako ostatni krok w sieci neuronowej.

  • Sigmoid – do klasyfikacji binarnej
  • Softmax – do klasyfikacji wieloklasowej

Mając na wejściu tensor z wartościami przed aktywacją, zastosuj odpowiednią funkcję aktywacji, aby uzyskać końcowy wynik.

torch.nn zostało już zaimportowane jako nn.

Instrukcje 1/2

undefined XP
  • 1
    • Utwórz funkcję sigmoid i zastosuj ją na input_tensor, aby wygenerować prawdopodobieństwo dla zadania klasyfikacji binarnej.
  • 2
    • Utwórz funkcję softmax i zastosuj ją na input_tensor, aby wygenerować prawdopodobieństwo dla zadania klasyfikacji wieloklasowej.