helper.py

import matplotlib.pyplot as plt
from IPython import display

plt.ion()

def plot(scores, mean_scores):
    plt.clf()
    plt.plot(scores, label='Score', color='blue', alpha=0.7, linewidth=1.5, linestyle='-')
    plt.plot(mean_scores, label='Rolling average', color='orange', alpha=0.7, linewidth=2, linestyle='-')
    plt.text(len(scores) - 1, scores[-1], f'{scores[-1]}', ha='center', va='bottom', fontsize=10, color='blue',
             weight='bold')
    plt.text(len(mean_scores) - 1, mean_scores[-1], f'{mean_scores[-1]:.2f}', ha='center', va='bottom', fontsize=10,
             color='orange', weight='bold')
    plt.legend(loc='upper left', shadow=True, fontsize='medium')
    plt.tick_params(axis='both', which='major', labelsize=8)
    plt.grid(True, linestyle='--', alpha=0.5)
    plt.title('Training Progress', fontsize=14)
    plt.xlabel('Number of Games', fontsize=10)
    plt.ylabel('Score', fontsize=10)
    plt.ylim(ymin=0)
    plt.tight_layout()
    display.clear_output(wait=True)
    display.display(plt.gcf())
    plt.show(block=False)
    plt.pause(.1)