Update SKMeans.hpp

Correct normalization dimension
flucoma · Nov 22, 2024 · 00061f4 · 00061f4
1 parent c76186c
commit 00061f4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/algorithms/public/SKMeans.hpp b/include/algorithms/public/SKMeans.hpp
@@ -34,7 +34,7 @@ class SKMeans : public KMeans
     using namespace _impl;
     assert(!mTrained || (dataset.pointSize() == mDims && mK == k));
     MatrixXd dataPoints =
-        asEigen<Matrix>(dataset.getData()).colwise().normalized();
+        asEigen<Matrix>(dataset.getData()).rowwise().normalized();
     MatrixXd dataPointsT = dataPoints.transpose();
     if (mTrained) { mAssignments = assignClusters(dataPointsT);}
     else