zalando-datasets/fashion_mnist
Viewer • Updated • 70k • 24.8k • 69
How to use Okneas/my_model with Keras:
# !pip install -U keras tensorflow huggingface_hub
# Keras needs TensorFlow installed to read "hf://" paths, so the tensorflow backend is selected here;
# "jax" and "torch" also work for computation once TensorFlow is installed.
import os
os.environ["KERAS_BACKEND"] = "tensorflow"
import keras
model = keras.saving.load_model("hf://Okneas/my_model")
Модель нейронной сети, представляет собой нейронную сеть с двумя сверточными слоями, двумя слоями пулинга, слоем дропаута (для предотвращения переобучения), полносвязным слоем и функцией активации softmax. Она была обучена на датасете Fashion MNIST, который содержит изображения одежды разных категорий. Модель определяет метку (одежда, обувь, сумка) по рисунку 28*28
Последовательная архитектура модели выглядит следующим образом:

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