1. Nauka
  2. /
  3. Kursy
  4. /
  5. Skalowalne modele AI z PyTorch Lightning

Connected

ćwiczenie

Podział danych za pomocą LightningDataModule

Uzupełnij metodę setup w klasie LightningDataModule. Odpowiedni podział zbioru danych sprawia, że model jest trenowany na jednym podzbiorze, a walidowany na innym – co pomaga unikać przetrenowania.

dataset został już wcześniej zaimportowany.

Instrukcje

100 XP
  • Zaimportuj random_split, aby podzielić zbiór danych na podzbiory treningowy i walidacyjny.
  • Podziel zbiór danych na podzbiór treningowy (80%) i walidacyjny (20%) za pomocą random_split.