caffe.help

Infogain Loss Layer

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

message InfogainLossParameter {
  // Specify the infogain matrix source.
  optional string source = 1;
  optional int32 axis = 2 [default = 1]; // axis of prob
}

Dialogue & Discussion