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 merupakan bagian dari kursus
Pengantar Deep Learning dengan PyTorch
Latihan interaktif langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
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)