1. Nauka
    2. /
  2. Kursy
    3. /
  3. Głębokie uczenie z PyTorch – poziom średnio zaawansowany
Connected

ćwiczenie

Ewaluacja modelu

Pętla treningowa jest już gotowa – model był trenowany przez 1000 epok i jest dostępny jako net. Skonfigurowano też test_dataloader w dokładnie taki sam sposób jak wcześniej train_dataloader – z tą różnicą, że dane są wczytywane z katalogu testowego, a nie treningowego.

Możesz teraz ocenić model na danych testowych. W tym celu napisz pętlę ewaluacyjną, która przejdzie przez kolejne partie danych testowych, pobierze predykcje modelu dla każdej z nich i obliczy wynik dokładności. Do dzieła!

Instrukcje

100 XP
  • Skonfiguruj metrykę ewaluacyjną jako Accuracy dla klasyfikacji binarnej i przypisz ją do acc.
  • Dla każdej partii danych testowych pobierz wyniki modelu i przypisz je do outputs.
  • Po zakończeniu pętli oblicz łączną dokładność na zbiorze testowym i przypisz ją do test_accuracy.