1 Star 0 Fork 1

陈梓华/classification

forked from lonerlin/classification 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
plt_record.py 772 Bytes
一键复制 编辑 原始数据 按行查看 历史
lonerlin 提交于 2021-09-15 10:29 . 添加一个画loss和acc的程序。
import matplotlib.pyplot as plt
import json
import numpy as np
dict = {}
with open("record.txt", 'r') as f:
dict = json.load(f)
train_record = dict['train']
val_record = dict['val']
fig, (ax1, ax2) = plt.subplots(2, 1, figsize=(12, 12))
ax1.plot(train_record['loss'], color='b', label="Training loss")
ax1.plot(val_record['loss'], color='r', label="validation loss")
ax1.set_xticks(np.arange(1, len(train_record['loss']), 1))
# ax1.set_yticks(np.arange(0, 1, 0.1))
ax2.plot(train_record['Acc1'], color='b', label="Training accuracy")
ax2.plot(val_record['Acc1'], color='r', label="Validation accuracy")
ax2.set_xticks(np.arange(1, len(train_record['loss']), 1))
legend = plt.legend(loc='best', shadow=True)
plt.tight_layout()
plt.savefig('record.jpg')
plt.show()
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/chen-zihua1536/classification.git
git@gitee.com:chen-zihua1536/classification.git
chen-zihua1536
classification
classification
master

搜索帮助

0d507c66 1850385 C8b1a773 1850385