机器学习之训练数据分析图像的绘制（python+csv）

# -*- coding:utf-8 -*-
import csv
import matplotlib.pyplot as plt

# 文件读取
file = open('./log.csv', encoding='utf-8-sig')
reader = csv.reader(file)

# 数据准备 并将字符型数据转换为浮点型数据
x = list(reader)
step =[]
loss = []
val_loss=[]
for i in range(1, len(x), 1):
    step.append(float(x[i][0]))
    loss.append(float(x[i][1]))
    val_loss.append(float(x[i][2]))

plt.plot(step, loss, marker='o', mec='r', mfc='w', label='loss')
plt.plot(step, val_loss, marker='*', ms=10, label='val_loss')
plt.legend()
plt.show()