Предсказание Кераса: возможно ли получить доступ к элементу вектора предсказания и вернуть его как скаляр? - PullRequest
Новая
       79

Предсказание Кераса: возможно ли получить доступ к элементу вектора предсказания и вернуть его как скаляр?

0 голосов
/ 28 сентября 2019

Я использую Keras для DNN для задачи выделения ресурсов, где размерность вывода DNN равна количеству ресурсов (скажем, есть R ресурсов).Каждый из R выходных узлов содержит пользователя, которому назначен ресурс (то есть, имея 10 пользователей, каждый выходной узел принимает целочисленное значение от 1 до 10).В функции потерь я хочу получить доступ к этим целым числам, но я не могу этого сделать.Моя функция потери состоит в следующем, где y_pred является выходом DNN: 

# Define custom loss
def custom_loss(num_outputs):

    # Create a loss function 
    def loss(y_true,y_pred):
        for k in range(num_outputs/2):            
            user_k = y_pred[0][k]
            user_k_integer = int(user_k)

#... return loss

Когда я пытаюсь скомпилировать модель, я получаю следующее сообщение об ошибке: Аргумент TypeError: int () должен быть строкойили число, а не 'Tensor'

То, что я хочу, это значение, хранящееся в y_pred [0] [k] в качестве скаляра, и затем использовать его как целое число для индексации.

Anybodyесть идеи как решить мою проблему?

Заранее большое спасибо!

...