wangwei created SINGA-91:
----------------------------
Summary: Add SoftmaxLayer and ArgSortLayer
Key: SINGA-91
URL: https://issues.apache.org/jira/browse/SINGA-91
Project: Singa
Issue Type: New Feature
Reporter: wangwei
These two layers are added to get prediction results (e.g., predicted labels),
which can be output by CSVOutputLayer.
SoftmaxLayer applies the Softmax function against its source layer to compute
its probability distribution over all labels.
ArgSortLayer sorts labels based on their scores (e.g., probability) in
descending order. It has a configuration field *argsort_conf*,
{code}
message ArgSortProto {
optional int32 topk = 1 [default = 1];
}
{code}
Topk labels will be kept.
ArgSortLayer cannot be used in the training phase, because it does not
implement the ComputeGradient function. It is used merely for extracting
prediction results. If it connects to a CSVOutputLayer, then topk labels will
be dumped into a csv file, topk labels per line.
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
