1.0 KiB
1.0 KiB
| title |
|---|
| Infogain Loss Layer |
Infogain Loss Layer
- Layer type:
InfogainLoss - Doxygen Documentation
- Header:
./include/caffe/layers/infogain_loss_layer.hpp - CPU implementation:
./src/caffe/layers/infogain_loss_layer.cpp
A generalization of MultinomialLogisticLossLayer that takes an "information gain" (infogain) matrix specifying the "value" of all label pairs.
Equivalent to the MultinomialLogisticLossLayer if the infogain matrix is the identity.
Parameters
- Parameters (
Parameter infogain_param) - From
./src/caffe/proto/caffe.proto:
{% highlight Protobuf %} {% include proto/InfogainLossParameter.txt %} {% endhighlight %}