用python将pt文件画图

单个文件直接画图：

# 将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()