1. Обучение
    2. /
  2. Курса
    3. /
  3. Введение в глубокое обучение с PyTorch
Connected

упражнение

Вычисление точности с помощью torchmetrics

Отслеживание точности в процессе обучения помогает определить эпоху с наилучшими результатами.

В этом упражнении вы будете использовать torchmetrics для вычисления точности на наборе данных о масках с тремя классами. Функция plot_errors выделит неправильно классифицированные примеры, что поможет проанализировать ошибки модели.

Пакет torchmetrics уже импортирован. Выходные данные модели outputs — это вероятности softmax, а labels — векторы в формате one-hot encoding.

Инструкции 1/2

undefined XP
    1
    2
  • Создайте метрику точности "multiclass" для трёх классов.
  • Вычислите точность для каждого батча загрузчика данных.