Как передать decimal_y_values ​​для обучения, где последний слой модели имеет 10 нейронов с активацией softmax, а потеря - Earth Mover Loss - PullRequest
Новая
       66

Как передать decimal_y_values ​​для обучения, где последний слой модели имеет 10 нейронов с активацией softmax, а потеря - Earth Mover Loss

0 голосов
/ 08 мая 2020

Каким должен быть формат / типы данных y-меток для обучения, если фактические y-метки cab представляют собой любое десятичное число от 0 до 9 (4.1,8.5 и c), а последний выходной слой определяется как: 

out = Dense(10,activation='softmax')(previous_layer)

и используемая функция потерь: 

def earth_mover_loss(y_true, y_pred):
    cdf_ytrue = K.cumsum(y_true, axis=-1)
    cdf_ypred = K.cumsum(y_pred, axis=-1)
    samplewise_emd = K.sqrt(K.mean(K.square(K.abs(cdf_ytrue - cdf_ypred)), axis=-1))
    return K.mean(samplewise_emd)

Как мне загружать метки в моей сети. Хочу мими c эту модель

...