用python将pt文件画图

拼搏现实的明天。 2023-10-08 22:00 10阅读 0赞

## 单个文件直接画图： ##

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