MulaiMulai sekarang secara gratis

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

Lihat Kursus

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