From 7ec132ca5e02a0d770a4e88d26dccc2e1e0c4c29 Mon Sep 17 00:00:00 2001
From: Omar Shrit <omar@avontech.fr>
Date: Sun, 3 Dec 2023 16:16:13 +0100
Subject: [PATCH] Change the centroids type to MatType instead of Mat<double>

Signed-off-by: Omar Shrit <omar@avontech.fr>
---
 src/mlpack/methods/dbscan/dbscan.hpp      | 4 ++--
 src/mlpack/methods/dbscan/dbscan_impl.hpp | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

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<typename MatType>
   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<typename MatType>
   size_t Cluster(const MatType& data,
                  arma::Row<size_t>& 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<typename RangeSearchType, typename PointSelectionPolicy>
 template<typename MatType>
 size_t DBSCAN<RangeSearchType, PointSelectionPolicy>::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<typename MatType>
 size_t DBSCAN<RangeSearchType, PointSelectionPolicy>::Cluster(
     const MatType& data,
     arma::Row<size_t>& assignments,
-    arma::mat& centroids)
+    MatType& centroids)
 {
   const size_t numClusters = Cluster(data, assignments);