Skip to content

AndrewRook/NFLWin

Repository files navigation

NFLWin

Build Status Documentation Status

Estimate Win Probability (WP) for plays in NFL games:

>>> import pandas as pd
>>> from nflwin.model import WPModel
>>> standard_model = WPModel.load_model()
>>> plays = pd.DataFrame({
... "quarter": ["Q1", "Q2", "Q4"],
... "seconds_elapsed": [0, 0, 600],
... "offense_team": ["NYJ", "NYJ", "NE"],
... "yardline": [-20, 20, 35],
... "down": [1, 3, 3],
... "yards_to_go": [10, 2, 10],
... "home_team": ["NYJ", "NYJ", "NYJ"],
... "away_team": ["NE", "NE", "NE"],
... "curr_home_score": [0, 0, 21],
... "curr_away_score": [0, 0, 10]
... })
>>> standard_model.predict_wp(plays)
array([ 0.58300397,  0.64321796,  0.18195466])

For full documentation, including information about methods and accuracy, click here.

License

MIT. See license file.