Skip to content

Generate predicted bracket from a kaggle march madness submission

License

Notifications You must be signed in to change notification settings

cshaley/bracketeer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

60 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

bracketeer

PyPI Python license Travis

Generate predicted bracket from a kaggle march madness machine learning competition submission. https://www.kaggle.com/c/march-machine-learning-mania-2017

Installation

To install, do one of the following two things:

pip install bracketeer

OR

git clone https://github.com/cshaley/bracketeer.git
cd bracketeer
python setup.py install

Usage:

from bracketeer import build_bracket
b = build_bracket(
        outputPath='output.png',
        teamsPath='sample/Teams.csv',
        seedsPath='sample/NCAATourneySeeds.csv',
        submissionPath='sample/submit_example.csv',
        slotsPath='sample/NCAATourneySlots.csv',
        resultsPath=None,
        year=2019
)

resultsPath is an optional CSV file containing results of games already played. If it is provided then in the cases that the team the user predicted to win loses, it will substitute the actual winning team. This is intended to visualize the predictions for later rounds, after the match-ups are set.

Example,

from bracketeer import build_bracket
b = build_bracket(
        outputPath='output_current.png',
        teamsPath='sample/Teams.csv',
        seedsPath='sample/NCAATourneySeeds.csv',
        submissionPath='sample/submit_example.csv',
        slotsPath='sample/NCAATourneySlots.csv',
        resultsPath='sample/current_results_example.csv',
        year=2019
)

Dependencies

  • binarytree
  • matplotlib
  • numpy
  • pandas
  • PIL

Additional input data/files not provided on kaggle:

  • empty_bracket.jpg - empty ncaa bracket
  • slot_coordinates.py - mapping dictionary from slots to image coordinates on empty_bracket.jpg
  • ordered_seed_list.py - order of seeds on the bracket

Output is a bracket filled in with team seeds, names, and winning likelihood for each game.
The empty bracket is shown below. alt tag

The predicted bracket (this program's output) is shown below, for a set of randomly generated predictions: alt tag

The predicted bracket, after applying known results for the first round (based on the first round of the 2019 tournament): alt tag

About

Generate predicted bracket from a kaggle march madness submission

Resources

License

Stars

Watchers

Forks

Packages

 
 
 

Languages