Mulai sekarangMulai 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 merupakan bagian dari kursus

Pengantar Deep Learning dengan PyTorch

Lihat Kursus

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

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