тензор - коробочный сюжет из тензора

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

Надеюсь, это поможет!