如何使用dlib进行神经网络回归?

时间:2016-12-03 15:42:59

标签: c++ neural-network regression dlib

似乎dlib需要一个损失层来决定如何处理距离输入层最远的层。我找不到任何关于损失层的文档,但似乎没有办法只有一些求和层。

总结最后一层的所有值正是我对回归所需要的(参见:https://deeplearning4j.org/linear-regression

我正在考虑编写自定义损失图层,但无法找到有关此内容的信息。

那么,我是否在这里监督了一些相应的层,或者是否有可能拥有我需要的东西?

1 个答案:

答案 0 :(得分:2)

dlib' s machine learning page的菜单中列出了dlib中的丢失层。寻找单词"损失层"。有很多文档。

目前发布的dlib版本不包含回归损失。但是,如果从github获取当前代码,则可以使用新的loss_mean_squared图层进行回归。请参阅:https://github.com/davisking/dlib/blob/master/dlib/dnn/loss_abstract.h