[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