Figure methods

Pre-written functions for generating figures.

Visualizations for bandit experiments.

bandit.figures.plot_average_rewards(reward_histories: List[T], axis: Optional[matplotlib.axes._axes.Axes] = None, **plot_kwargs) → Tuple[matplotlib.figure.Figure, matplotlib.axes._axes.Axes]

Given a set of reward histories, plot their averages.

TODO: add **plot_kwargs

Parameters:

• reward_histories (List) – a list of received rewards for all bandits
• axis (mpl.axes.Axes) – axis to use for plotting, default None
• plot_kwargs (dict) – key-value pairs for the axis.plot function

bandit.figures.plot_reward_distributions(rewards: List[T], axis: Optional[matplotlib.axes._axes.Axes] = None) → Tuple[matplotlib.figure.Figure, matplotlib.axes._axes.Axes]

Create violin plots of the reward distributions.

TODO: add **plot_kwargs

Parameters:

• rewards (List[Rewards]) – rewards to make distributions of
• axis (mpl.axes.Axes) –