У меня есть функция, которая выдает мне значение в виде массива в качестве вывода при запуске, но мне нужен вывод в виде целочисленного процента для части результатов
def pred_datsci(file_path):
prev_precompute = learn.precompute
learn.precompute = False
try:
trn_tfms, val_tfms = tfms_from_model(arch,sz)
test_img = open_image(file_path)
im = val_tfms(test_img)
pred = learn.predict_array(im[None])
class_index = (np.exp(pred))
class_index1 = np.argmax(np.exp(pred))
print(class_index*100)
return data.classes[class_index1]
finally:
learn.precompute = prev_precompute
Вот как выглядит результат:
pred_datsci(f"data/dogscats1/valid/dogs/12501.jpg")
![enter image description here](https://i.stack.imgur.com/D33U0.png)
У меня вопрос, как мне отобразить эти два значения как:
Кошка% = 15,81724%
Собака% = 84,18274%