Question

我正试图在概率（y轴）和花名（x轴）两个变量之间创建一个框。概率是张量。对于花的名称，我必须从字典（flower_dict）中选择名称，该字典中的键是从另一个张量（类Index）引用的。如何创建箱形图？感谢您的帮助

print("Probability:", probs)
Probability: tensor([[ 0.9961,  0.0020,  0.0011,  0.0005,  0.0001]], device='cuda:0')

print("Class Index:", classes)
Class Index: tensor([[  21,  3,  45,  34,  27]], device='cuda:0')

print(flower_dict)
{'21': 'fire lily', '3': 'canterbury bells', '45': 'bolero deep blue', '1': 'pink primrose', '34': 'mexican aster', '27': 'prince of wales feathers', '7': 'moon orchid', '16': 'globe-flower', '25': 'grape hyacinth', '26': 'corn poppy', '79': 'toad lily', '39': 'siam tulip', '24': 'red ginger'}

Answer 1

我猜您的意思是在此处创建条形图（通常使用箱形图来描述诸如检查特征等的分布；例如在鸢尾花的情况下，您可能希望在箱形图中检查隔片的长度）。如果您想要条形图，则可以尝试以下代码：

import numpy as np
import matplotlib.pyplot as plt

y = probs.flatten().numpy()
x = [flower_dict[str(i)] for i in classes.flatten().numpy()]

p1 = plt.bar(x, y, width)

plt.ylabel('Probability')
plt.xticks(ind, x)

plt.show()

希望这会有所帮助！

张量-张量的箱形图

1 个答案: