Я новичок в tensorflow.js и машинном обучении в целом. Я пытался создать базовую модель tf.js для набора данных titanic на kaggle, и у меня получилось довольно далеко, но я столкнулся с ошибкой при попытке ...

1
Praneel Samal 20 Дек 2020 в 01:30

1 ответ

Лучший ответ

Ошибка oneHot возникает из-за функции потерь sparseCategoricalCrossentropy. Вы используете категориальную функцию потерь, которая ожидает вывода нескольких длин, но ваша модель выводит вектор длины 1 (последний слой - model.add(tf.layers.dense({units: 1}));). Вместо этого вы должны вывести столько узлов, сколько есть категорий, которые вы классифицируете, поэтому 2 узла, если вы делаете метку выживания / отсутствия выживания.

0
Andy K 30 Дек 2020 в 06:24