Skip to content

Commit 00061f4

Browse files
authored
Update SKMeans.hpp
Correct normalization dimension
1 parent c76186c commit 00061f4

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

include/algorithms/public/SKMeans.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ class SKMeans : public KMeans
3434
using namespace _impl;
3535
assert(!mTrained || (dataset.pointSize() == mDims && mK == k));
3636
MatrixXd dataPoints =
37-
asEigen<Matrix>(dataset.getData()).colwise().normalized();
37+
asEigen<Matrix>(dataset.getData()).rowwise().normalized();
3838
MatrixXd dataPointsT = dataPoints.transpose();
3939
if (mTrained) { mAssignments = assignClusters(dataPointsT);}
4040
else

0 commit comments

Comments
 (0)