MulaiMulai sekarang secara gratis

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

Lihat Kursus

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)
Edit dan Jalankan Kode