diff --git a/src/mlpack/methods/dbscan/dbscan.hpp b/src/mlpack/methods/dbscan/dbscan.hpp index de3d20f04b..f411dba01d 100644 --- a/src/mlpack/methods/dbscan/dbscan.hpp +++ b/src/mlpack/methods/dbscan/dbscan.hpp @@ -80,7 +80,7 @@ class DBSCAN */ template size_t Cluster(const MatType& data, - arma::mat& centroids); + MatType& centroids); /** * Performs DBSCAN clustering on the data, returning number of clusters @@ -108,7 +108,7 @@ class DBSCAN template size_t Cluster(const MatType& data, arma::Row& assignments, - arma::mat& centroids); + MatType& centroids); private: //! Maximum distance between two points to be part of same cluster. diff --git a/src/mlpack/methods/dbscan/dbscan_impl.hpp b/src/mlpack/methods/dbscan/dbscan_impl.hpp index bb240ccd5d..8ad8e1f047 100644 --- a/src/mlpack/methods/dbscan/dbscan_impl.hpp +++ b/src/mlpack/methods/dbscan/dbscan_impl.hpp @@ -43,7 +43,7 @@ template template size_t DBSCAN::Cluster( const MatType& data, - arma::mat& centroids) + MatType& centroids) { // These assignments will be thrown away, but there is no way to avoid // calculating them. @@ -62,7 +62,7 @@ template size_t DBSCAN::Cluster( const MatType& data, arma::Row& assignments, - arma::mat& centroids) + MatType& centroids) { const size_t numClusters = Cluster(data, assignments);