<p>I do some change on the api of FineTuneGradient, this api may provide better optimization codes</p>
<pre><code>class SoftmaxFineTune
{
public:
template<typename T>
static void LastGradient(arma::mat const &input,
arma::mat const &weights,
T const &model,
arma::mat &gradient)
{
gradient = (weights.t() * model.Probabilities()) /
static_cast<double>(input.n_cols);
gradient = gradient % (input % (1 - input));
}
static void Gradient(arma::mat const &input,
arma::mat const &weight,
arma::mat const &deriv,
arma::mat &output)
{
output = (weight.t() * deriv) % (input % (1 - input));
}
};
</code></pre>
<p style="font-size:small;-webkit-text-size-adjust:none;color:#666;">—<br>Reply to this email directly or <a href="https://github.com/mlpack/mlpack/issues/458#issuecomment-146073692">view it on GitHub</a>.<img alt="" height="1" src="https://github.com/notifications/beacon/AJ4bFBzuYXeoBowBuNxv_byJb_gMCSbRks5o5JSpgaJpZM4GIjrs.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/458#issuecomment-146073692"></link>
<meta itemprop="name" content="View Issue"></meta>
</div>
<meta itemprop="description" content="View this Issue on GitHub"></meta>
</div>