plot.py

import matplotlib.pyplot as plt
from IPython import display

plt.ion()

# To plot the agent's performance over time
def plot(scores, mean_scores, epsilon):
    display.clear_output(wait=True)
    display.display(plt.gcf())
    plt.clf()
    if epsilon > 0:
        plt.title(f'Episode: {len(scores)}  Epsilon: {epsilon:.2f} (Exploration Phase)')
    else:
        plt.title(f'Episode: {len(scores)}  Epsilon: {epsilon:.2f} (Exploitation Phase)')
    plt.xlabel('Number of Games')
    plt.ylabel('Score')
    plt.plot(scores)
    plt.plot(mean_scores)
    plt.ylim(ymin=0)
    plt.text(len(scores)-1, scores[-1], f'{scores[-1]}')
    plt.text(len(mean_scores)-1, mean_scores[-1], f'{mean_scores[-1]}')
    plt.show(block=False)
    plt.pause(.1)