Membangun pengklasifikasi biner di PyTorch
Ingat bahwa jaringan saraf kecil dengan satu lapis linear yang diikuti fungsi sigmoid adalah pengklasifikasi biner. Perilakunya sama seperti regresi logistik.
Latihlah membangun jaringan kecil ini dan menafsirkan keluaran dari pengklasifikasi tersebut.
Latihan ini adalah bagian dari kursus
Pengantar Deep Learning dengan PyTorch
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
import torch
import torch.nn as nn
input_tensor = torch.Tensor([[3, 4, 6, 2, 3, 6, 8, 9]])
# Implement a small neural network for binary classification
model = nn.Sequential(
nn.____(____),
nn.____()
)
output = model(input_tensor)
print(output)