• 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 (Parameter infogain_param)
• From ./src/caffe/proto/caffe.proto:

{% highlight Protobuf %} {% include proto/InfogainLossParameter.txt %} {% endhighlight %}