[mlpack-svn] r16787 - mlpack/trunk/src/mlpack/core/tree/binary_space_tree
fastlab-svn at coffeetalk-1.cc.gatech.edu
fastlab-svn at coffeetalk-1.cc.gatech.edu
Tue Jul 8 08:56:01 EDT 2014
Author: rcurtin
Date: Tue Jul 8 08:56:01 2014
New Revision: 16787
Log:
Return the correct type of the matrix, because it isn't necessarily dense.
Modified:
mlpack/trunk/src/mlpack/core/tree/binary_space_tree/binary_space_tree.hpp
Modified: mlpack/trunk/src/mlpack/core/tree/binary_space_tree/binary_space_tree.hpp
==============================================================================
--- mlpack/trunk/src/mlpack/core/tree/binary_space_tree/binary_space_tree.hpp (original)
+++ mlpack/trunk/src/mlpack/core/tree/binary_space_tree/binary_space_tree.hpp Tue Jul 8 08:56:01 2014
@@ -280,9 +280,9 @@
size_t& SplitDimension() { return splitDimension; }
//! Get the dataset which the tree is built on.
- const arma::mat& Dataset() const { return dataset; }
+ const MatType& Dataset() const { return dataset; }
//! Modify the dataset which the tree is built on. Be careful!
- arma::mat& Dataset() { return dataset; }
+ MatType& Dataset() { return dataset; }
//! Get the metric which the tree uses.
typename BoundType::MetricType Metric() const { return bound.Metric(); }
More information about the mlpack-svn
mailing list