Набор данных Титаник. Логиты и метки различной формы ошибки - PullRequest
0 голосов
/ 02 мая 2018

В настоящее время я создаю DNN для классификации наборов данных Titanic. Во время этого процесса я получаю следующую ошибку:

InvalidArgumentError: logits and labels must have the same first dimension, got logits shape [1,5] and labels shape [5]

Мой код можно найти здесь:

https://github.com/isaac-altair/Titanic-Dataset/blob/master/Titanic%2BProject.ipynb

Я попытался изменить форму логитов, используя np.reshape (logits, (-1, 1)), но это не сработало.

Спасибо

...