Himpunan data dua masukan
Membangun model multi-masukan dimulai dengan membuat himpunan data kustom yang dapat memasok semua masukan ke model. Pada latihan ini, Anda akan membangun himpunan data Omniglot yang menyajikan triplet yang terdiri dari:
- Gambar karakter yang akan diklasifikasikan,
- Vektor alfabet one-hot berdimensi 30, bernilai nol di semua posisi kecuali satu yang menandai ID alfabet asal karakter,
- Label target, sebuah bilangan bulat antara 0 dan 963.
Anda disediakan samples, sebuah daftar 3-tuple yang memuat path berkas gambar, vektor alfabetnya, dan label target. Selain itu, impor berikut sudah disiapkan untuk Anda, jadi mari kita mulai!
from PIL import Image
from torch.utils.data import DataLoader, Dataset
from torchvision import transforms
Latihan ini adalah bagian dari kursus
Deep Learning Lanjutan dengan PyTorch
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
class OmniglotDataset(Dataset):
def __init__(self, transform, samples):
# Assign transform and samples to class attributes
____ = ____
____ = ____