You are viewing a plain text version of this content. The canonical link for it is here.
Posted to issues@systemml.apache.org by "Mike Dusenberry (JIRA)" <ji...@apache.org> on 2017/06/15 18:25:01 UTC
[jira] [Updated] (SYSTEMML-1678) Add new 1D & 2D top_k utility
functions
[ https://issues.apache.org/jira/browse/SYSTEMML-1678?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel ]
Mike Dusenberry updated SYSTEMML-1678:
--------------------------------------
Summary: Add new 1D & 2D top_k utility functions (was: Add new 1D & 2D predict class utility functions)
> Add new 1D & 2D top_k utility functions
> ---------------------------------------
>
> Key: SYSTEMML-1678
> URL: https://issues.apache.org/jira/browse/SYSTEMML-1678
> Project: SystemML
> Issue Type: Sub-task
> Reporter: Mike Dusenberry
> Assignee: Mike Dusenberry
>
> We should add new {{predict_class}} and {{predict_class2d}} utility functions (in {{nn/util.dml}}) that accept a matrix {{X}} and return a matrix {{out}} with the predicted classes based on the max probability. For the 1D case, {{X}} is the output of a {{softmax}} layer, and thus each row will contain a set of normalized class probabilities
--
This message was sent by Atlassian JIRA
(v6.4.14#64029)