Прогнозы и индексы Tensorflow Grab для значений выше пороговых значений

Каков самый простой способ захватить соответствующие значения и индексы предсказания, основанные на значениях выше определенного порога?

Рассмотрим эту проблему:

sess = tf.InteractiveSession() predictions = tf.constant([[ 0.32957435, 0.82079124, 0.54503286, 0.51966476, 0.63359714, 0.92034972, 0.13774526, 0.45154464, 0.18284607, 0.14604568], [ 0.78612137, 0.98291659, 0.4841609 , 0.63260579, 0.21568334, 0.82978213, 0.05054879, 0.09517837, 0.28309393, 0.01788473], [ 0.05706763, 0.24366784, 0.04608512, 0.32987678, 0.2342416 , 0.91725373, 0.60084391, 0.51787591, 0.74161232, 0.30830121], [ 0.67310858, 0.6250236 , 0.42477703, 0.37107778, 0.65123832, 0.97282803, 0.59533679, 0.49564457, 0.54935825, 0.63008392], [ 0.70233917, 0.48129809, 0.59114349, 0.63535333, 0.71188867, 0.4799161 , 0.90896237, 0.86089945, 0.47896886, 0.83451629], [ 0.82923532, 0.8950938 , 0.99231505, 0.05526769, 0.98151541, 0.18153167, 0.63851702, 0.07426929, 0.91846335, 0.81246626], [ 0.12850153, 0.23018432, 0.29871917, 0.71228445, 0.13235569, 0.41061044, 0.98215759, 0.90024149, 0.53385031, 0.92247963], [ 0.87011361, 0.44218826, 0.01772344, 0.87317121, 0.52231467, 0.86476815, 0.25352192, 0.31709731, 0.38249743, 0.74694788], [ 0.15262914, 0.49544573, 0.49644637, 0.07461977, 0.13706958, 0.18619633, 0.86163998, 0.03700352, 0.51173556, 0.40018845]]) score_idx = tf.where(predictions > 0.8) scores = tf.SparseTensor(prediction_idx, tf.gather_nd(predictions, score_idx), dense_shape=tf.shape(predictions, out_type=tf.int64)) scores.eval() 

Я могу легко получить разреженный тензор, который имеет все прогнозы выше 0,8, но в конечном итоге я хочу вернуть два отдельных 1D-тензора:

  • Прогнозируемые индексы = список индексов выше порога (например, 0,8)
  • Оценки = баллы для соответствующих примеров

Итак, для первой строки, которая:

 [ 0.32957435, 0.82079124, 0.54503286, 0.51966476, 0.63359714, 0.92034972, 0.13774526, 0.45154464, 0.18284607, 0.14604568] 

Я хочу вернуться:

  • preded_indices = [1,5]
  • оценки = [0,821, 0,920]

Есть ли простое решение, которое мне не хватает?

 
Interesting Posts for Van-Lav

Как получить среднее изображение из 100 изображений с помощью PIL?

Python: Как получить ссылку на модуль внутри самого модуля?

Как распределить ось x двух подзаговоров после их создания?

Большинство Pythonic способ распечатать * не более * некоторое количество десятичных знаков

Numpy – loadtxt и используемый конвертер для даты

Как подсчитать по два с Python 'range'

Запустить скрипт bash с python – TypeError: bufsize должен быть целым числом

Localhost перестает обновляться, когда запускаются различные скрипты flask / python, как я могу это исправить?

boost.python выставить функцию, которая возвращает вектор <MyClass>

AttributeError: объект «NoneType» не имеет атрибута «endswith»

Как сортировать огромные файлы с Python?

Есть ли ограничение на количество таблиц, которые может иметь база данных PostgreSQL?

Обновление Python Matplotlib

В чем разница между списком и списком ?

Python vs Bash – В каких задачах каждый из них превосходит другие по производительности?

Python - лучший язык программирования в мире.