关于二分类

来源：3-4 卷积神经网络实战

Ams0

2019-03-23

老师，课中的卷积神经网络是进行多分类的，如果只想进行二分类，课中的代码哪些地方需要改动呢？

写回答

1回答

正十七

2019-04-09

已采纳

其实可以不用改什么，只需要让神经网络的最后一个全连接层输出为2就可以了。然后保证你的数据中的y取值为0或者1.

y_ = tf.layers.dense(flatten, 2 (之前是10))

Ams0

感谢老师！已经解决了！

2019-04-10

共1条回复

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学，深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

相似问题

回答 1

回答 1

回答 1

回答 1

回答 1