用python将pt文件画图

最新推荐文章于 2024-08-09 17:08:17 发布

秦羽

最新推荐文章于 2024-08-09 17:08:17 发布

阅读量1.4k

点赞数 1

分类专栏： # ECG学习笔记文章标签： python numpy 机器学习

本文链接：https://blog.youkuaiyun.com/qq_43585922/article/details/129763019

版权

ECG学习笔记专栏收录该内容

2 篇文章

订阅专栏

这段代码展示了如何使用Python的torch和matplotlib库从.pt文件加载数据，并进行数据可视化。首先，它加载单个文件中的Loss数据并绘制折线图。接着，同一张图上展示了Loss、Accuracy和F1三个指标。最后，对于每个指标，分别创建了单独的图表进行展示。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

单个文件直接画图：

# 将pt文件画图展示。已经过测试

import torch
import matplotlib.pyplot as plt

# 从.pt文件中加载数据
data = torch.load('../Loss.pt')

# 打印数据
print("Loss:")

print(data)

# 将数据转换为numpy数组
data = data.numpy()

# 绘制折线图
plt.plot(data)

# 添加标题和轴标签
plt.title('Loss Plot')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')

# 显示图形
plt.show()

多个文件一起画图

# 将pt文件在一张图中展示。已经过测试

import torch
import matplotlib.pyplot as plt

# 从.pt文件中加载数据
Loss = torch.load('../experiments/24_03_2023__10_06_20ECGCNN_S_physio_net_dataset_challange_two_classes/metrics/Loss.pt')
Accuracy = torch.load('../experiments/24_03_2023__10_06_20ECGCNN_S_physio_net_dataset_challange_two_classes/metrics/Accuracy.pt')
F1 = torch.load('../experiments/24_03_2023__10_06_20ECGCNN_S_physio_net_dataset_challange_two_classes/metrics/F1.pt')

# 打印数据
print("Loss:")
print(Loss)

print("Accuracy:")
print(Accuracy)

print("F1:")
print(F1)

# 将数据转换为numpy数组
Loss = Loss.numpy()
Accuracy = Accuracy.numpy()
F1 = F1.numpy()


# 绘制折线图
plt.plot(Loss, label='Loss')
plt.plot(Accuracy, label='Accuracy')
plt.plot(F1, label='F1')

# 添加标题和轴标签
plt.title('ECGCNNBiLSTM_B_S Plot')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')

# 显示图形
plt.show()

多个文件分别画图：

# 将多个pt文件画图展示。已经经过测试

import torch
import matplotlib.pyplot as plt

# 从.pt文件中加载数据
Loss = torch.load('../experiments/24_03_2023__21_46_47ECGCNN_S_physio_net_dataset_challange_two_classes/metrics/Loss.pt')
Accuracy = torch.load('../experiments/24_03_2023__21_46_47ECGCNN_S_physio_net_dataset_challange_two_classes/metrics/Accuracy.pt')
F1 = torch.load('../experiments/24_03_2023__21_46_47ECGCNN_S_physio_net_dataset_challange_two_classes/metrics/F1.pt')

# 打印数据
print("Loss:")
print(Loss)

print("Accuracy:")
print(Accuracy)

print("F1:")
print(F1)

# 将数据转换为numpy数组
Loss = Loss.numpy()
Accuracy = Accuracy.numpy()
F1 = F1.numpy()


arr  = {
    'Loss': Loss,
    'Accuracy': Accuracy,
    'F1': F1
}

for a in arr:
    # 绘制折线图
    plt.plot(arr[a])
    # 添加标题和轴标签:LOSS
    plt.title('ECGBiCLSTM_B_S' + a + 'Plot')
    plt.xlabel('X-axis')
    plt.ylabel('Y-axis')
    # 显示图形
    plt.show()