2016-07-11 2 views
0

Я пытаюсь написать пользовательскую целевую функцию для keras, и мне нужно манипулировать данными в y_true, y_pred, чтобы сделать это.keras пользовательская целевая функция, получение данных в y_pred, y_true?

def custom_objective(y_true, y_pred): 
    print y_true, y_pred 

Когда я печатаю эти значения, я получаю:

dense_1_target Softmax.0 

Я ожидаю, с партии размером 128 и SoftMax с двумя классами, y_true и y_pred быть в форме (128, 2), Но это не так.

Как мне добраться до данных здесь?

ответ

Смежные вопросы