Împărțirea Datelor în Train, Val și Test cu random_split în PyTorch & Deep Learning Practic

Sintaxă & invocare Python
train, val = torch.utils.data.random_split(dataset, [0.8, 0.2])

Funcția torch.utils.data.random_split() fragmentează un dataset în sub-mulțimi disjuncte (clasa Subset) conform fracțiunilor sau numărului absolut de mostre specificate.

import torch
from torch.utils.data import random_split

# Generator cu seed fix pentru reproductibilitate științifică
generator = torch.Generator().manual_seed(42)

# Împărțire 80% antrenament, 10% validare, 10% testare
train_ds, val_ds, test_ds = random_split(
    full_dataset,
    [0.8, 0.1, 0.1],
    generator=generator
)

print(f"Mostre Train: {len(train_ds)}, Val: {len(val_ds)}, Test: {len(test_ds)}")

Sub-mulțimile create rețin referința la setul de date original și o listă de indici aleatori, fără a duplica matricele în memorie.

Învață AI • Comunitate & Q&A

Întrebări & Exemple practice: Împărțirea Datelor în Train, Val și Test cu random_split

Ai o întrebare despre Împărțirea Datelor în Train, Val și Test cu random_split sau ai descoperit un truc, prompt sau fragment de cod util? Distribuie-l cu ceilalți membri ai comunității!

🤖

Fii primul care pune o întrebare sau distribuie un fragment de cod / prompt!

Nu au fost adăugate încă întrebări sau resurse pentru această pagină. Dacă ai testat un prompt sau ai un snippet util, adaugă-l mai jos pentru comunitate.

Comunitate AI Adaugă o întrebare sau distribuie un cod / prompt

Moderare & Siguranță:

Pentru a menține calitatea și securitatea informațiilor, fiecare întrebare sau fragment de cod/prompt este analizat (manual sau prin moderare asistată de AI) înainte de publicare.