graph.py

import matplotlib.pyplot as plt
import numpy as np
import json

FileName = 'log'

def read_json(path = './log.json'):
    file = open(path, 'r')
    return json.load(file)

def plot_chainer_log(json_obj, y_keys, linestyles, colors, x_key='epoch'):
    dict = {}
    x = []
    for key in y_keys:
        dict[key] = []
    for obj in json_obj:
        x.append(obj[x_key])
        for key in y_keys:
            dict[key].append(obj[key])
    for key, sty, clr in zip(y_keys, linestyles, colors):
        plt.plot(x, dict[key], label=key, color=clr, linestyle=sty)
    plt.xlabel = x_key

axes = plt.gca()
#axes.set_ylim([0, 1.0])
plt.grid(which='major', color='gray', linestyle='-')
plt.grid(which='minor', color='gray', linestyle='-')
plt.legend(loc='lower right')
plot_chainer_log(read_json(FileName + '.json'), y_keys=['main/loss', 'validation/main/loss'], linestyles=['solid', 'solid'], colors=['black', 'darkgray'])
plt.legend(loc='upper right')
#plt.show()
plt.savefig(FileName + '.png')