CommencezCommencez gratuitement

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

Voir le cours

Instructions de l’exercice

  • Entraînez le modèle avec seed_diff comme variable d'entrée et score_diff comme 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)
Modifier et exécuter le code