Antrenarea unui model de rețea neuronală pe date despre articole vestimentare
În acest exercițiu, vei antrena rețeaua neuronală complet conectată pe care ai construit-o în exercițiul anterior, folosind date de imagine. Datele de antrenament sunt disponibile în două variabile: train_data, care conține datele despre pixeli pentru 50 de imagini din cele trei clase vestimentare, și train_labels, care conține reprezentările codificate one-hot ale etichetelor pentru fiecare dintre aceste 50 de imagini. Transformă datele în formatul de intrare așteptat de rețea, apoi antrenează modelul pe datele și etichetele de antrenament.
Modelul model pe care l-ai compilat în exercițiul anterior, împreună cu train_data și train_labels, sunt disponibile în spațiul tău de lucru.
Acest exercițiu face parte din cursul
Modelarea imaginilor cu Keras
Instrucțiuni pentru exercițiu
- Pregătește datele pentru antrenare prin redimensionarea lor.
- Antrenează modelul pasând datele de intrare și etichetele de antrenament metodei
.fit()a modelului.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# Reshape the data to two-dimensional array
train_data = train_data.reshape(____, ____)
# Fit the model
model.fit(____, ____, validation_split=0.2, epochs=3)