socceraction.atomic.vaep.features¶

Implements the feature tranformers of the VAEP framework.

socceraction.atomic.vaep.features.actiontype_onehot(actions)¶

Get the one-hot-encoded type of each action.

socceraction.atomic.vaep.features.direction(actions)¶

Get the direction of the action as components of the unit vector.

Parameters:: actions (Actions) – The actions of a game.
Returns:: The x-component (‘dx’) and y-compoment (‘mov_angle’) of the unit vector of each action.
Return type:: Features

socceraction.atomic.vaep.features.feature_column_names(fs, nb_prev_actions=3)¶

Return the names of the features generated by a list of transformers.

Parameters:

fs (list(callable)) – A list of feature transformers.
nb_prev_actions (int, default=3 # noqa: DAR103) – The number of previous actions included in the game state.

Returns:

The name of each generated feature.

Return type:

list(str)

socceraction.atomic.vaep.features.goalscore(gamestates)¶

Get the number of goals scored by each team after the action.

Parameters:: gamestates (GameStates) – The gamestates of a game.
Returns:: The number of goals scored by the team performing the last action of the game state (‘goalscore_team’), by the opponent (‘goalscore_opponent’), and the goal difference between both teams (‘goalscore_diff’).
Return type:: Features

socceraction.atomic.vaep.features.location(actions)¶

Get the location where each action started.

socceraction.atomic.vaep.features.movement_polar(actions)¶

Get the distance covered and direction of each action.

Parameters:: actions (Actions) – The actions of a game.
Returns:: The distance covered (‘mov_d’) and direction (‘mov_angle’) of each action.
Return type:: Features

socceraction.atomic.vaep.features.play_left_to_right(gamestates, home_team_id)¶

Perform all action in the same playing direction.

This changes the start and end location of each action, such that all actions are performed as if the team plays from left to right.

Parameters:

Returns:

The game states with all actions performed left to right.

Return type:

list(pd.DataFrame)

socceraction.atomic.vaep.features.polar(actions)¶

Get the polar coordinates of each action’s start location.

The center of the opponent’s goal is used as the origin.