Навчання моделі нейронної мережі на даних про одяг
У цій вправі ви навчите повнозв'язну нейронну мережу, яку зібрали в попередній вправі, на даних зображень. Тренувальні дані подано у двох змінних: train_data, що містить піксельні дані для 50 зображень трьох класів одягу, і train_labels, що містить one-hot кодування міток для кожного з цих 50 зображень. Перетворіть дані у формат, який очікує мережа, а потім навчіть модель на тренувальних даних і мітках.
model, який ви скомпілювали в попередній вправі, а також train_data і train_labels доступні у вашому робочому середовищі.
Ця вправа є частиною курсу
Моделювання зображень у Keras
Інструкції до вправи
- Підготуйте дані до навчання, змінивши їхню форму.
- Навчіть модель, передавши тренувальні вхідні дані та мітки до методу моделі
.fit().
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
# Reshape the data to two-dimensional array
train_data = train_data.reshape(____, ____)
# Fit the model
model.fit(____, ____, validation_split=0.2, epochs=3)