[mlpack-svn] r16800 - mlpack/trunk/src/mlpack/methods/perceptron
fastlab-svn at coffeetalk-1.cc.gatech.edu
fastlab-svn at coffeetalk-1.cc.gatech.edu
Wed Jul 9 15:55:15 EDT 2014
Author: rcurtin
Date: Wed Jul 9 15:55:15 2014
New Revision: 16800
Log:
Don't use arma::unique() because it's slow.
Modified:
mlpack/trunk/src/mlpack/methods/perceptron/perceptron_impl.hpp
Modified: mlpack/trunk/src/mlpack/methods/perceptron/perceptron_impl.hpp
==============================================================================
--- mlpack/trunk/src/mlpack/methods/perceptron/perceptron_impl.hpp (original)
+++ mlpack/trunk/src/mlpack/methods/perceptron/perceptron_impl.hpp Wed Jul 9 15:55:15 2014
@@ -33,10 +33,8 @@
const arma::Row<size_t>& labels,
int iterations)
{
- arma::Row<size_t> uniqueLabels = arma::unique(labels);
-
WeightInitializationPolicy WIP;
- WIP.Initialize(weightVectors, uniqueLabels.n_elem, data.n_rows + 1);
+ WIP.Initialize(weightVectors, arma::max(labels) + 1, data.n_rows + 1);
// Start training.
classLabels = labels;
More information about the mlpack-svn
mailing list