CNN 模型训练结果可视化

import pandas as pd
import matplotlib.pyplot as plt

# 读取log.csv文件
df = pd.read_csv('log.csv', header=None, skiprows=146, nrows=70)
df.columns = ['Epoch', 'Accuracy', 'Loss']

# 绘制精确度图
plt.figure(figsize=(10, 5))
plt.plot(df['Epoch'], df['Accuracy'], 'o-', color='b', label='Accuracy')
plt.xlabel('Epoch')
plt.ylabel('Accuracy')
plt.title('Accuracy vs. Epoch')
plt.legend(loc='upper left')
plt.savefig('accuracy.jpg')
plt.show()

# 绘制损失函数图
plt.figure(figsize=(10, 5))
plt.plot(df['Epoch'], df['Loss'], 'o-', color='r', label='Loss')
plt.xlabel('Epoch')
plt.ylabel('Loss')
plt.title('Loss vs. Epoch')
plt.legend(loc='upper left')
plt.savefig('loss.jpg')
plt.show()