Skip to content

mcmckinley/DraftStarsModel

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 

Repository files navigation

Draft Stars - Model

Creates, trains and exports a model which predicts the outcome of Brawl Stars matches.

Created with PyTorch


Input

To run this program, you need a dataset of Brawl Stars matches. You can collect one yourself with another repo of mine: DraftStarsDataCollection


Each match in the dataset contains the following values:

  • IDs for all six brawlers1
    • There are six players in each match (3 versus 3). The first three IDs for each match refer to the left (or blue) team. The last three refer to the right (or red) team.
    • Very important note: there are no brawlers with indices 33 and 55. This is because the official brawler list, when requested from Supercell's API, does not include brawlers at these indices.
  • Which team won
    • This column is titled 'did_blue_team_win'. 1 means yes and a 0 means no (the team on right, or red team, won.)
  • Trophy counts of each player on that brawler
    • This is useful for weighing the statistical significance of any given match. When a low skill team wins, we assume that their team composition was superior.

I would strongly recommmend initializing the brawler and map embedding layers with manually chosen values. With Google Colab this notebook is already set up to import spreadsheets of initialization values that I have created myself. If you wish to run these locally, you should download my spreadsheets or make your own (would not recommend, it takes a while.)

Brawler Data Spreadsheet:

Map Data Spreadsheet:


Output

  • The pytorch model
  • Embedding layers for the brawlers and maps, in CSV format
    • Why CSV? For one, it's readable. It's interesting to see how the model takes the initialization data and tweaks it. More importantly however is the fact that we can alter it ourselves.
  • A list of maps, in the order than the model interprets them. Without this the backend won't know which map is selected.

About

Creates an AI model which predicts the outcome of Brawl Stars matches.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors