Entraîner le modèle sur les données de basketball du tournoi
Maintenant que le modèle est compilé, vous êtes prêt à l'entraîner sur des données!
Dans cet exercice, vous utiliserez un jeu de données de pointages provenant de matchs du tournoi de basketball universitaire américain. Chaque ligne du jeu de données contient les identifiants des équipes : team_1 et team_2, sous forme d'entiers. On y trouve aussi la différence de classement entre les équipes (les classements sont attribués par le comité du tournoi et représentent la force des équipes) ainsi que la différence de pointage du match (p. ex., si team_1 gagne par 5 points, la différence de pointage est 5).
Pour entraîner le modèle, vous fournissez une matrice de variables X (dans ce cas, une seule colonne : la différence de classement) et une matrice de variables Y (ici aussi une seule colonne : la différence de pointage).
Le DataFrame games_tourney, ainsi que l'objet model déjà compilé, sont disponibles dans votre environnement de travail.
Cette activité fait partie du cours
Machine Learning avancé avec Keras
Instructions de l’exercice
- Entraînez le modèle avec
seed_diffcomme variable d'entrée etscore_diffcomme variable de sortie. - Utilisez 1 époque, une taille de lot de 128 et une fraction de validation de 10 %.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Now fit the model
____(games_tourney_train[____], games_tourney_train[____],
epochs=____,
batch_size=____,
validation_split=____,
verbose=True)