plot_losses.py

import matplotlib.pyplot as plt
import numpy as np


def plot(file):
    try:
        plt.style.use('ggplot')
        data = np.loadtxt(f"{file}.csv", delimiter=',', skiprows=0)
        fig, ax = plt.subplots(figsize=(16, 9))

        x = np.arange(data.shape[0])
        ax.plot(x, data[:, 0], label='Loss')
        ax.legend()

        plt.savefig(f"{file}.png", bbox_inches='tight')
        plt.close()
    except Exception as e:
        print(e)


filename = 'data/saves/%s/results/%s'    # without extension

# for i in range(17):
#     plot(filename % (i, 'training_loss'))
#
# plot(filename % (17, 'training_loss-0'))
# plot(filename % (17, 'training_loss-1'))

#for i in range(18, 30):
#    plot(filename % (i, 'training_loss'))
#
#plot('data/saves/34/results/training_loss')
#plot('data/saves/10-bis/results/training_loss')
#plot('data/saves/35/after-1-game/results/training_loss')
#for i in range(2, 6):
#    plot(f'data/saves/35/after-{i}-games/results/training_loss')

for i in range(4):
    plot(plot(filename % (36, f'training_loss-{i}')))