MulaiMulai sekarang secara gratis

Fungsi sigmoid dan softmax

Fungsi sigmoid dan softmax adalah fungsi aktivasi kunci dalam deep learning, yang sering digunakan sebagai langkah akhir pada neural network.

  • Sigmoid untuk klasifikasi biner
  • Softmax untuk klasifikasi multi-kelas

Diberikan sebuah tensor keluaran pra-aktivasi dari sebuah network, terapkan fungsi aktivasi yang sesuai untuk memperoleh keluaran akhir.

torch.nn sudah diimpor sebagai nn.

Latihan ini adalah bagian dari kursus

Pengantar Deep Learning dengan PyTorch

Lihat Kursus

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

input_tensor = torch.tensor([[2.4]])

# Create a sigmoid function and apply it on input_tensor
sigmoid = nn.____()
probability = ____(____)
print(probability)
Edit dan Jalankan Kode