[
https://issues.apache.org/jira/browse/SINGA-91?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Sheng Wang resolved SINGA-91.
-----------------------------
Resolution: Fixed
> 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)
