MulaiMulai sekarang secara gratis

Dari regresi ke klasifikasi multi-kelas

Model yang Anda lihat untuk klasifikasi biner, klasifikasi multi-kelas, dan regresi semuanya serupa, dengan beberapa penyesuaian pada model.

Mulailah membangun model untuk regresi, lalu ubah model tersebut agar melakukan klasifikasi multi-kelas.

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, 7, 10, 12, 2, 3, 6, 8, 9]])

# Implement a neural network with exactly four linear layers
model = nn.____(
  ____
)

output = model(input_tensor)
print(output)
Edit dan Jalankan Kode