Antrenează modelul pe datele turneului de baschet
Acum că modelul este compilat, ești pregătit să îl antrenezi pe date!
În acest exercițiu, vei folosi un set de date cu scorurile meciurilor din turneul universitar de baschet din SUA. Fiecare rând din set de date conține identificatorii echipelor: team_1 și team_2, ca valori întregi. De asemenea, conține diferența de seed dintre echipe (seed-urile sunt atribuite de comitetul turneului și reprezintă un clasament al puterii echipelor) și diferența de scor a meciului (de exemplu, dacă team_1 câștigă cu 5 puncte, diferența de scor este 5).
Pentru a antrena modelul, furnizezi o matrice de variabile X (în acest caz o singură coloană: diferența de seed) și o matrice de variabile Y (în acest caz o singură coloană: diferența de scor).
DataFrame-ul games_tourney, împreună cu obiectul model compilat, sunt disponibile în spațiul tău de lucru.
Acest exercițiu face parte din cursul
Deep Learning avansat cu Keras
Instrucțiuni pentru exercițiu
- Antrenează modelul cu
seed_diffca variabilă de intrare șiscore_diffca variabilă de ieșire. - Folosește 1 epocă, un batch size de 128 și o fracție de validare de 10%.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# Now fit the model
____(games_tourney_train[____], games_tourney_train[____],
epochs=____,
batch_size=____,
validation_split=____,
verbose=True)