У меня есть CNN, где выходным измерением является [Нет, 10]
Это проблема с несколькими метками, где выходные данные указывают на возможные категории, которым может принадлежать x.(например, изображение может быть классифицировано как cat
dark
и т. д.)
Что у меня сейчас есть, как я могу изменить код на версию keras?Я не могу найти эквивалент sigmoid_cross_entropy_with_logits
model = tf.layers.dense(L3, category_num, activation=None)
cross_entropy = tf.nn.sigmoid_cross_entropy_with_logits(logits=model, labels=Y)
cost = tf.reduce_mean(tf.reduce_sum(cross_entropy, axis=1))
optimizer = tf.train.AdamOptimizer(learning_rate).minimize(cost)