[mlpack-git] master: 1 : fix style 2 : remove useless header 3 : reduce temporary variable (54c7f22)

gitdub at mlpack.org gitdub at mlpack.org
Wed Apr 20 02:46:09 EDT 2016


Repository : https://github.com/mlpack/mlpack
On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/ebf77f8b13323a87c433b6f639deb2369188b00c...b08ae02b90e18f97366b236e7a4d8725cd6e9050

>---------------------------------------------------------------

commit 54c7f22a6ddaf68cca491ad098cc27324740a8fa
Author: stereomatchingkiss <stereomatchingkiss at gmail.com>
Date:   Wed Apr 20 14:46:09 2016 +0800

    1 : fix style
    2 : remove useless header
    3 : reduce temporary variable


>---------------------------------------------------------------

54c7f22a6ddaf68cca491ad098cc27324740a8fa
 src/mlpack/core/util/split_data.hpp | 25 ++++++++++---------------
 1 file changed, 10 insertions(+), 15 deletions(-)

diff --git a/src/mlpack/core/util/split_data.hpp b/src/mlpack/core/util/split_data.hpp
index 04fc6f0..1ddd3a7 100644
--- a/src/mlpack/core/util/split_data.hpp
+++ b/src/mlpack/core/util/split_data.hpp
@@ -3,11 +3,7 @@
 
 #include <mlpack/core.hpp>
 
-#include <iterator>
-#include <numeric>
-#include <random>
 #include <tuple>
-#include <vector>
 
 namespace mlpack {
 namespace util {
@@ -34,26 +30,25 @@ namespace util {
  *@endcode
  */
 template<typename T, typename U>
-void TrainTestSplit(arma::Mat<T> const &input,
-                    arma::Row<U> const &inputLabel,
+void TrainTestSplit(const arma::Mat<T> &input,
+                    const arma::Row<U> &inputLabel,
                     arma::Mat<T> &trainData,
                     arma::Mat<T> &testData,
                     arma::Row<U> &trainLabel,
                     arma::Row<U> &testLabel,
-                    double const testRatio)
+                    const double testRatio)
 {
   size_t const testSize =
       static_cast<size_t>(input.n_cols * testRatio);
-  size_t const trainSize = input.n_cols - testSize;
+  const size_t trainSize = input.n_cols - testSize;
   trainData.set_size(input.n_rows, trainSize);
   testData.set_size(input.n_rows, testSize);
   trainLabel.set_size(trainSize);
   testLabel.set_size(testSize);
 
-  using Col = arma::Col<size_t>;
-  Col const sequence = arma::linspace<Col>(0, input.n_cols - 1,
-                                           input.n_cols);
-  arma::Col<size_t> const order = arma::shuffle(sequence);
+  const arma::Col<size_t> order =
+      arma::shuffle(arma::linspace<arma::Col<size_t>>(0, input.n_cols - 1,
+                                                      input.n_cols));
 
   for(size_t i = 0; i != trainSize; ++i)
   {
@@ -84,9 +79,9 @@ void TrainTestSplit(arma::Mat<T> const &input,
 template<typename T,typename U>
 std::tuple<arma::Mat<T>, arma::Mat<T>,
 arma::Row<U>, arma::Row<U>>
-TrainTestSplit(arma::Mat<T> const &input,
-               arma::Row<U> const &inputLabel,
-               double const testRatio)
+TrainTestSplit(const arma::Mat<T> &input,
+               const arma::Row<U> &inputLabel,
+               const double testRatio)
 {
   arma::Mat<T> trainData;
   arma::Mat<T> testData;




More information about the mlpack-git mailing list