如何使神经网络具有回归和分类输出?

时间:2019-04-17 15:16:03

标签: keras

我已经训练了CNN用于带有回归输出的转向角预测。我还为CNN进行了分类培训(两节课:道路,无道路)。但是我想把它结合起来。怎么做?训练新模型不是问题。

1 个答案:

答案 0 :(得分:1)

您将需要更新y值以同时具有类别和角度,并编写自定义损失函数以添加回归损失和类别损失。