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 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, 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)