drawWER.py

import matplotlib.pyplot as plt
#import sys
import argparse

parser = argparse.ArgumentParser('Draw WER')
parser.add_argument('flag', type=str, help='si/no with/without testing')
#parser.add_argument('folder', type=str, help='pred_logs_layer_1 or others')
args = parser.parse_args()
flag = args.flag
#base = args.folder+'/'
base = 'pred_logs/'
#if len(sys.argv) != 2:
#    print('Usage: python3 drawCER.py si/no (with or without testing)')
#    exit()
#flag = sys.argv[1]
#base = 'pred_logs/'

cer = open(base+'wer_train.log', 'r')
cer_data = cer.read().split(' ')[:-1]
cerr = [float(i) for i in cer_data]

cer_v = open(base+'wer_valid.log', 'r')
cer_data_v = cer_v.read().split(' ')[:-1]
cerr_v = [float(i) for i in cer_data_v]

if flag == 'si':
    cer_t = open(base+'wer_test.log', 'r')
    cer_data_t = cer_t.read().split(' ')[:-1]
    cerr_t = [float(i) for i in cer_data_t]

plt.plot(cerr, 'r-')
cer_spot, = plt.plot(cerr, 'ro')

plt.plot(cerr_v, 'b-')
cer_spot_v, = plt.plot(cerr_v, 'bo')

if flag == 'si':
    plt.plot(cerr_t, 'g-')
    cer_spot_t, = plt.plot(cerr_t, 'go')
    plt.legend([cer_spot, cer_spot_v, cer_spot_t], ['WER train', 'WER valid', 'WER test'])
else:
    plt.legend([cer_spot, cer_spot_v], ['CER train', 'CER valid'])

plt.xlabel('epoch')
plt.ylim(0, 1)
plt.title('word error rate')
plt.grid(color='m', linestyle='--', linewidth=0.5)
plt.show()

cer.close()
cer_v.close()
if flag == 'si':
    cer_t.close()