溫馨提示×

pytorch畫圖如何繪制餅狀圖

小樊
90
2024-12-25 15:56:19
欄目: 深度學習

在PyTorch中,您可以使用matplotlib庫來繪制餅狀圖。以下是一個簡單的示例:

首先,確保已經安裝了matplotlib庫。如果沒有安裝,可以使用以下命令安裝:

pip install matplotlib

接下來,您可以使用以下代碼繪制餅狀圖:

import torch
import matplotlib.pyplot as plt

# 創建一個包含分類概率的張量
probabilities = torch.tensor([0.1, 0.2, 0.3, 0.4])

# 繪制餅狀圖
plt.pie(probabilities.numpy(), labels=['Class 1', 'Class 2', 'Class 3', 'Class 4'], autopct='%1.1f%%')

# 設置圖表標題
plt.title("Pie Chart Example")

# 顯示圖表
plt.show()

在這個例子中,我們首先導入了所需的庫,然后創建了一個包含分類概率的PyTorch張量。接著,我們使用plt.pie()函數繪制餅狀圖,并通過autopct參數顯示每個類別的百分比。最后,我們設置了圖表標題并使用plt.show()顯示圖表。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女