<p>Hi <a href="https://github.com/zoq" class="user-mention">@zoq</a> <br>
Sorry for the late response but I have been working on this issue with vasanthkalingeri. I would like to complete this. Considering your above comment, I plan to use the MultiClassClassificationLayer as it returns the the output parameter. Would it be okay to use it or would you recommend writing another layer for the same?</p>

<p style="font-size:small;-webkit-text-size-adjust:none;color:#666;">&mdash;<br />You are receiving this because you are subscribed to this thread.<br />Reply to this email directly or <a href="https://github.com/mlpack/mlpack/issues/414#issuecomment-210071230">view it on GitHub</a><img alt="" height="1" src="https://github.com/notifications/beacon/AJ4bFDd1asPR8dCAOuiosmgJ1vux32L8ks5p3nxAgaJpZM4DnsTV.gif" width="1" /></p>
<div itemscope itemtype="http://schema.org/EmailMessage">
<div itemprop="action" itemscope itemtype="http://schema.org/ViewAction">
  <link itemprop="url" href="https://github.com/mlpack/mlpack/issues/414#issuecomment-210071230"></link>
  <meta itemprop="name" content="View Issue"></meta>
</div>
<meta itemprop="description" content="View this Issue on GitHub"></meta>
</div>