penaltyblog.models.ZeroInflatedPoissonGoalsModel#

class penaltyblog.models.ZeroInflatedPoissonGoalsModel(goals_home: Sequence[int] | ndarray[Any, dtype[int64]] | Series, goals_away: Sequence[int] | ndarray[Any, dtype[int64]] | Series, teams_home: Sequence[str] | ndarray[Any, dtype[str_]] | Series, teams_away: Sequence[str] | ndarray[Any, dtype[str_]] | Series, weights: float | Sequence[float] | ndarray[Any, dtype[float64]] | Series | None = None)[source]#

Bases: BaseGoalsModel

Zero-Inflated Poisson Model for Football Goal Scoring

__init__(goals_home: Sequence[int] | ndarray[Any, dtype[int64]] | Series, goals_away: Sequence[int] | ndarray[Any, dtype[int64]] | Series, teams_home: Sequence[str] | ndarray[Any, dtype[str_]] | Series, teams_away: Sequence[str] | ndarray[Any, dtype[str_]] | Series, weights: float | Sequence[float] | ndarray[Any, dtype[float64]] | Series | None = None)[source]#

Initialises the ZeroInflatedPoissonGoalsModel class.

Parameters:

  • goals_home (array_like) – The number of goals scored by the home team

  • goals_away (array_like) – The number of goals scored by the away team

  • teams_home (array_like) – The names of the home teams

  • teams_away (array_like) – The names of the away teams

  • weights (array_like, optional) – The weights of the matches, by default None

Methods

__init__(goals_home, goals_away, teams_home, ...)

Initialises the ZeroInflatedPoissonGoalsModel class.

fit()

Fits the model to the data.

get_params()

Returns the model's fitted parameters as a dictionary

load(filepath)

Loads a model from a file.

predict(home_team, away_team[, max_goals])

Predicts the probability of each scoreline for a given match.

save(filepath)

Saves the model to a file using pickle.

Attributes

params

Property to retrieve the fitted model parameters.
fit()[source]#

Fits the model to the data.

Must be implemented by the subclass.

get_params() → Dict[str, float][source]#

Returns the model’s fitted parameters as a dictionary

Returns:

A dict containing the model’s parameters

Return type:

dict

classmethod load(filepath: str) → Any#

Loads a model from a file.

Parameters:

filepath (str) – The path to the file from which the model will be loaded.

Returns:

An instance of the model.

Return type:

Any

property params: dict#

Property to retrieve the fitted model parameters. Same as get_params(), but allows attribute-like access.

Returns:

A dictionary containing attack, defense, home advantage, and correlation parameters.

Return type:

dict

Raises:

ValueError – If the model has not been fitted yet.

predict(home_team: str, away_team: str, max_goals: int = 15) → FootballProbabilityGrid[source]#

Predicts the probability of each scoreline for a given match.

Must be implemented by the subclass.

save(filepath: str)#

Saves the model to a file using pickle.

Parameters:

filepath (str) – The path to the file where the model will be saved.