Maison >développement back-end >Tutoriel Python >Prédire la chimie des joueurs NBA à l'aide de réseaux de neurones graphiques
Bonjour à tous, je m'appelle sea_turt1e.
Cet article partagera le processus et les résultats de la création d'un modèle d'apprentissage automatique pour prédire l'alchimie des joueurs dans la Ligue nationale de basket-ball (NBA), un sport que j'aime beaucoup.
Pour les lecteurs qui ne connaissent pas la NBA, certaines parties de cet article peuvent être difficiles à comprendre. La « réaction chimique » peut être comprise d’un point de vue plus intuitif. De plus, bien que cet article se concentre sur la NBA, la méthode pourrait également être appliquée à d’autres sports et même à la prédiction de la chimie interpersonnelle.
Regardons d'abord les résultats des prédictions. J'entrerai plus en détail sur l'ensemble de données et les détails techniques plus tard.
Dans la prédiction des réactions chimiques, les bords rouges indiquent de bonnes réactions chimiques, les bords noirs indiquent des réactions chimiques modérées et les bords bleus indiquent de mauvaises réactions chimiques.
La fraction sur le côté représente le score de réaction chimique, allant de 0 à 1.
Voici les prédictions de chimie pour les joueurs vedettes. Le graphique ne contient que des paires de joueurs n'ayant jamais joué pour la même équipe.
En regardant les prédictions des joueurs vedettes qui n’ont jamais joué ensemble, les résultats ne sont pas toujours intuitifs.
Par exemple, LeBron James et Stephen Curry ont fait preuve d'une excellente coordination aux Jeux olympiques, ce qui indique une bonne alchimie. D’un autre côté, Nikola Jokic devrait avoir une mauvaise alchimie avec les autres joueurs.
Pour rapprocher les prédictions de la réalité, j'ai testé l'alchimie entre les joueurs dans des échanges réels pour la saison 2022-23.
Étant donné que les données de la saison 2022-23 ne sont pas incluses dans les données d'entraînement, les prédictions qui correspondent à des impressions réalistes peuvent indiquer l'efficacité du modèle.
Plusieurs échanges importants auront lieu au cours de la saison 2022-23.
Voici les pronostics pour les joueurs clés dont Kevin Durant, Kyrie Irving et Rui Hachimura.
Les prédictions chimiques pour leur nouvelle équipe sont les suivantes :
Ces résultats semblent assez précis compte tenu de la dynamique de la saison 2022-23. (Même si les choses ont changé pour les Suns et les Mavericks la saison suivante.)
Ensuite, j'expliquerai les aspects techniques, y compris le framework GNN et la préparation des ensembles de données.
GNN (Graph Neural Network) est un réseau conçu pour traiter des données structurées sous forme de graphiques.
Dans ce modèle, les « réactions chimiques entre joueurs » sont représentées sous forme d'arêtes de graphique, et le processus d'apprentissage est le suivant :
Pour les bords négatifs, le modèle donne la priorité aux « coéquipiers avec de faibles passes décisives » et affaiblit l'influence des « joueurs d'équipes différentes ».
AUC (aire sous la courbe) fait référence à l'aire sous la courbe ROC et est utilisée comme mesure pour évaluer les performances du modèle.
Plus l’AUC est proche de 1, plus la précision est élevée. Dans cette étude, l'ASC du modèle était d'environ 0,73, soit un résultat moyen à supérieur à la moyenne.
Voici la courbe d'apprentissage et la progression de l'AUC au cours du processus de formation :
La principale innovation réside dans la construction du jeu de données.
Pour quantifier l’alchimie, je suppose que « passes décisives élevées » signifie une bonne alchimie. Sur la base de cette hypothèse, l'ensemble de données est structuré comme suit :
De plus, les coéquipiers avec un faible nombre de passes décisives sont explicitement considérés comme ayant une mauvaise alchimie.
Tous les codes sont disponibles sur GitHub.
En suivant les instructions du README, vous devriez être en mesure de reproduire le processus de formation et de tracer les graphiques décrits ici.
https://www.php.cn/link/867079fcaff2dfddeb29ca1f27853ef7
Il y a encore place à l'amélioration et je prévois d'atteindre les objectifs suivants :
Dans cet article, je décris mes tentatives pour prédire la chimie des joueurs NBA.
Bien que le modèle soit encore en développement, j'espère obtenir des résultats plus intéressants avec de nouvelles améliorations.
Bienvenue pour laisser vos réflexions et suggestions dans la zone de commentaires !
Si vous avez besoin d'améliorations supplémentaires, faites-le-moi savoir !
Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!