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


def data_load(path_list):
    x_train = []
    y_train = []
    h = 0
    with open(path_list, mode='r', encoding='utf-8-sig', newline='') as f:
        reader = csv.reader(f)
        for row in reader:  # 打开csv文件读取一行
            h += 1
            if h != 1:
                x_train.append(float(row[7]))
            # print(row)
    return x_train




loss1 = data_load('training_Bi.csv')
print(loss1)
loss2 = data_load('training_CB_Bi.csv')
print(loss2)
loss3 = data_load('training_s_Bi.csv')
print(loss3)


plt.figure(figsize=(20, 10), dpi=100)
plt.plot(loss1)
plt.plot(loss2, 'r--')
plt.plot(loss3, 'g-.')
plt.grid('on')
# plt.title('embedding+BiLSTM+CRF accuracy')
plt.ylabel('recall')
plt.xlabel('Epoch')
plt.legend(['Embedding+BiLSTM+CRF', 'COBW+BiLSTM+CRF', 'Skip-gram+BiLSTM+CRF'], loc='upper left')
plt.show()