Что означает output_dim в Keras Dense?

Question

Я новичок в Deep Learning. Я изучаю эту статью ссылка .

Я не могу понять, что означают output_dim=128 и output_dim=1 ???

Я ожидаю в качестве выходных данных в окончательном Dense количество классов = 2 (Кошка / Собака).

Кроме того, откуда 128 ???

Answer 1

Output_dim - размерность плотного вложения.

На выбор 128 в

classifier.add(Dense(output_dim = 128, activation = 'relu'))

совершенно произвольно, оно просто указывает размер полностью связанного слоя, который вы предпочитаете. Вы можете изменить его на другой номер.

1 в

classifier.add(Dense(output_dim = 1, activation = 'sigmoid'))

связано с проблемой бинарной классификации, нам просто нужна вероятность различить 2 группы. Если вероятность составляет не менее 0,5, мы классифицируем ее как собаку, если она меньше 0,5, мы классифицируем ее как кошку.

Если вы предпочитаете, вы также можете установить функцию активации равной softmax и output_dim равной 2 в качестве последнего слоя, хотя это не улучшит производительность.