my_model / README.md
Wauplin's picture
Wauplin HF staff
Set `library_name` to `tf-keras`.
d5c757b verified
|
raw
history blame
2 kB
metadata
datasets:
  - fashion_mnist
language:
  - ru
library_name: tf-keras
metrics:
  - accuracy
pipeline_tag: image-to-text

Описание модели

Модель нейронной сети, представляет собой нейронную сеть с двумя сверточными слоями, двумя слоями пулинга, слоем дропаута (для предотвращения переобучения), полносвязным слоем и функцией активации softmax. Она была обучена на датасете Fashion MNIST, который содержит изображения одежды разных категорий. Модель определяет метку (одежда, обувь, сумка) по рисунку 28*28

Последовательная архитектура модели

Последовательная архитектура модели выглядит следующим образом: Alt Text

Общее количество обучаемых параметров модели

Alt Text

Алгоритмы оптимизации и функция ошибки

Модель была обучена с помощью алгоритма оптимизации Adam и функции ошибки categorical_crossentropy.

Размеры тренировочного, валидационного и тестового датасетов

Размеры тренировочного, валидационного и тестового датасетов:

  • Размер тренировочного датасета: 60 000 изображений
  • Размер валидационного датасета: 10 000 изображений
  • Размер тестового датасета: 10 000 изображений

Результаты обучения модели

Alt Text

Работа модели

Alt Text